HDU 4735 舞蹈链可重复覆盖

点击打开链接

题意：给一个树和边权，然后树上的每个节点都有一个男孩或女孩，若有一个男孩的位置与一个女孩的位置的距离小于等于D，则这个女孩安全，现在可以换任意两个节点的人，问最少换多少次所有女孩都安全

思路：我们需要处理的就是每个节点如果是男孩他可以保护的所有的女孩，然后建舞蹈链的模版，最后当找到一组可行解时，比较其中节点是女孩的最少的那组解就是答案，而程序中还可以用这个值来减枝，强大！！！

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=810;
int L[maxn*maxn],R[maxn*maxn],U[maxn*maxn],D[maxn*maxn];//节点的上下左右四个方向的链表
int C[maxn*maxn],H[maxn],cnt[maxn],vis[maxn],A[maxn],Row[maxn],ans[maxn];//C列H行cnt列链表中元素个数
int n,m,id,fans,lpl;
int dis[maxn][maxn];
void init(int lll){
    for(int i=0;i<=lll;i++){
        cnt[i]=0;U[i]=D[i]=i;
        L[i+1]=i;R[i]=i+1;
    }
    R[lll]=0;id=lll+1;fans=0;
    memset(H,-1,sizeof(H));
}
void Link(int r,int c){
    cnt[c]++;C[id]=c;Row[id]=r;
    U[id]=U[c];D[U[c]]=id;
    D[id]=c;U[c]=id;
    if(H[r]==-1) H[r]=L[id]=R[id]=id;
    else{
        L[id]=L[H[r]];R[L[H[r]]]=id;
        R[id]=H[r];L[H[r]]=id;
    }
    id++;
}
void Remove(int Size){
    for(int j=D[Size];j!=Size;j=D[j])
        L[R[j]]=L[j],R[L[j]]=R[j];
}
void Resume(int Size){
    for(int j=D[Size];j!=Size;j=D[j])
        L[R[j]]=R[L[j]]=j;
}
int h(){
    int sum=0;
    memset(vis,0,sizeof(vis));
    for(int i=R[0];i;i=R[i]){
        if(vis[i]) continue;
        sum++;
        for(int j=D[i];j!=i;j=D[j]){
            for(int k=R[j];k!=j;k=R[k])
                vis[C[k]]=1;
        }
    }
    return sum;
}
void Dance(int k){
    int mm=maxn,pos,sum=0;
    if(k+h()>fans) return;
    for(int i=0;i<k;i++) sum+=A[ans[i]];
    if(k-sum>=lpl) return ;
    if(!R[0]){
        lpl=k-sum;
        return;
    }
    for(int i=R[0];i;i=R[i])
        if(mm>cnt[i]) mm=cnt[i],pos=i;
    for(int i=D[pos];i!=pos;i=D[i]){
        ans[k]=Row[i];
        Remove(i);
        for(int j=R[i];j!=i;j=R[j]) Remove(j);
        Dance(k+1);
        for(int j=R[i];j!=i;j=R[j]) Resume(j);
        Resume(i);
    }
}
void floyd(){
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
}
int main(){
    int T,u,v,co,cas=1;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        init(n);
        for(int i=1;i<=n;i++) scanf("%d",&A[i]),fans+=A[i];
        memset(dis,inf,sizeof(dis));
        for(int i=1;i<=n;i++) dis[i][i]=0;
        for(int i=0;i<n-1;i++) scanf("%d%d%d",&u,&v,&co),dis[u][v]=dis[v][u]=co;
        floyd();
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(dis[i][j]<=m) Link(i,j);
            }
        }
        lpl=inf;
        Dance(0);
//        cout<<fans<<"==fans"<<endl;
//        cout<<flag<<"==flag"<<endl;
        printf("Case #%d: ",cas++);
        if(lpl==inf) printf("-1\n");
        else printf("%d\n",lpl);
    }
    return 0;
}