hdu 4735 Little Wish~ lyrical step~(DLX)

题目链接：hdu 4735 Little Wish~ lyrical step~

题意：

有n个节点的树，每个节点可能是男孩，可能是女孩，节点之间有距离，现在要让所有的女孩周围距离D之内有男孩，问最小需要交换多少次男孩和女孩的位置。

题解：

把每个节点对小于D距离的全部link起来，然后DLX爆艹，意义就是选n个节点去覆盖全部节点，如果这个节点是女生，那么就是要替换的点。然后不断更新答案。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;

int boys,a[110],an;

struct DLX{
    const static int mn=1000;
    #define FF(i,A,s) for(int i=A[s];i!=s;i=A[i])
    #define INF 0x3f3f3f3f
    int L[mn],R[mn],U[mn],D[mn];
    int size,col[mn],row[mn],s[mn],H[mn];
    bool vis[70];
    int ans[mn],cnt;
    void init(int m){
        F(i,0,m)L[i]=i-1,R[i]=i+1,U[i]=D[i]=i,s[i]=0;
        memset(H,-1,sizeof(H));
        L[0]=m;R[m]=0;size=m+1;
    }
    void link(int r,int c){
         U[size]=c;D[size]=D[c];U[D[c]]=size;D[c]=size;
         if(H[r]<0)H[r]=L[size]=R[size]=size;
         else L[size]=H[r],R[size]=R[H[r]],L[R[H[r]]]=size,R[H[r]]=size;
         s[c]++,col[size]=c,row[size]=r,size++;
     }
    void del(int c){
        L[R[c]]=L[c];R[L[c]]=R[c];
        FF(i,D,c)FF(j,R,i)U[D[j]]=U[j],D[U[j]]=D[j],--s[col[j]];
    }
    void add(int c){
        R[L[c]]=L[R[c]]=c;
        FF(i,U,c)FF(j,L,i)++s[col[U[D[j]]=D[U[j]]=j]];
    }
    bool dfs(int k){//精确覆盖
        if(!R[0]){
            cnt=k;return 1;
        }
        int c=R[0];FF(i,R,0)if(s[c]>s[i])c=i;
        del(c);
        FF(i,D,c){
            FF(j,R,i)del(col[j]);
            ans[k]=row[i];if(dfs(k+1))return 1;
            FF(j,L,i)add(col[j]);
        }
        add(c);
        return 0;
    }
    void remove(int c){FF(i,D,c)L[R[i]]=L[i],R[L[i]]=R[i];}//重复覆盖
    void resume(int c){FF(i,U,c)L[R[i]]=R[L[i]]=i;}
    int A(){//估价函数
        int res=0;
        memset(vis,0,sizeof(vis));
        FF(i,R,0)if(!vis[i]){
                res++;vis[i]=1;
                FF(j,D,i)FF(k,R,j)vis[col[k]]=1;
            }
        return res;
    }
    void dfs(int now,int cnt){//重复覆盖
        if(now+A()>boys||cnt>=an)return;
        if(!R[0])
        {
            an=cnt;
            return;
        }
        int temp=INF,c;
        FF(i,R,0)if(temp>s[i])temp=s[i],c=i;
        FF(i,D,c){
            ans[now]=row[i];
            remove(i);FF(j,R,i)remove(j);
            dfs(now+1,cnt+!a[row[i]]);
            FF(j,L,i)resume(j);resume(i);
        }
    }
}dlx;

int t,n,d,cas;

int g[60],v[110],w[110],nxt[110],ed;

void adg(int x,int y,int z){v[++ed]=y,w[ed]=z,nxt[ed]=g[x],g[x]=ed;}

void dfs(int x,int rt,int fa=0,int dis=0)
{
    dlx.link(rt,x);
    for(int i=g[x];i;i=nxt[i])
        if(v[i]!=fa&&dis+w[i]<=d)
            dfs(v[i],rt,x,dis+w[i]);
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&d);
        boys=ed=0,dlx.init(n);
        memset(g,0,sizeof(g));
        F(i,1,n)scanf("%d",a+i),boys+=a[i];
        F(i,1,n-1)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            adg(x,y,z);
            adg(y,x,z);
        }
        F(i,1,n)dfs(i,i);
        an=INT_MAX;
        dlx.dfs(0,0);
        printf("Case #%d: %d
",++cas,an==INT_MAX?-1:an);
    }
    return 0;
}