【UVA1151】Buy or Build(MST+子集枚举)

传送门

Buy or Build
    题意：给出n个城市和一些模型，每一个模型可以将若干个不同的城市连通，需要一定费用。两两城市之间修建道路，费用为两城市欧几里得距离平方。求利用模型+修建道路使得n个城市连通的最小花费。

I think

    先求一遍原图MST，得到最小生成树边集E。子集枚举选择模型后再利用E中的边求MST。
    原因是在Kruskal过程中，以前被舍弃的边选择模型后依然会被舍弃。
    关于子集枚举：子集枚举的三种方法 http://www.cnblogs.com/itlqs/p/5720784.html
    贴一份自己的输出子集的代码

#include<cstdio>
int n,a[60];
int main() {
    scanf("%d",&n);
    for(int i=0;i<(1<<n);++i) {
        printf("{ ");
        for(int j=0;j<n;++j)
            if(i&(1<<j)) printf("%d ",j);
        printf("}\n");
    }
    return 0;
}

Code

#include<cstdio>
#include<algorithm>

using namespace std;
typedef long long LL;
const int sm = 1e3+10;

LL Ans,Ansx;
int N,Q,T,tot,Ct;
int X[sm],Y[sm];
int Fa[sm],Que[10][sm],c[10],t[10];
struct Edge{
    int Fm,To,L;
    bool operator < (const Edge&a) const{ 
        return L<a.L; 
    }
}_Ed[sm*sm>>1],Ed[sm];

void read(int &x) {
    char ch=getchar();x=0;
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
LL Min(LL x,LL y) { return x<y?x:y; }
void init() { 
    Ans=tot=Ct=0; 
}

int Find(int x) { return x==Fa[x]?x:Fa[x]=Find(Fa[x]); }

int Dist(int x,int y,int a,int b) {
    return (x-a)*(x-a)+(y-b)*(y-b);
}

LL Kruskal() {
    LL ret=0;
    for(int i=1,x,y;i<=tot;++i) {
        x=Find(Ed[i].Fm);
        y=Find(Ed[i].To);
        if(x!=y) 
            Fa[x]=y,ret+=Ed[i].L;
    }
    return ret;
}
int main() {
    read(T);
    while(T--) {
        init(),read(N),read(Q);
        for(int i=0;i<Q;++i) {
            read(t[i]),read(c[i]);
            for(int j=1;j<=t[i];++j)
                read(Que[i][j]);
        }
        for(int i=1;i<=N;++i)
            read(X[i]),read(Y[i]);
        for(int i=1;i<N;++i)
            for(int j=i+1;j<=N;++j)
                _Ed[++Ct]=(Edge){ i,j,Dist(X[i],Y[i],X[j],Y[j]) };
        sort(_Ed+1,_Ed+Ct+1);

        for(int i=1;i<=N;++i) Fa[i]=i;
        for(int i=1,x,y;i<=Ct;++i) {
            x=Find(_Ed[i].Fm); 
            y=Find(_Ed[i].To);
            if(x!=y) {
                Fa[x]=y,Ans+=_Ed[i].L;
                Ed[++tot]=_Ed[i];
                if(tot==N-1) break;
            }
        }   
        for(int S=0;S<(1<<Q);++S) {
            Ansx=0;
            for(int i=1;i<=N;++i) Fa[i]=i;
            for(int i=0;i<Q;++i)
                if(S&(1<<i)) {
                    Ansx+=c[i];
                    for(int j=2;j<=t[i];++j)
                        Fa[Find(Que[i][j])]=Find(Que[i][j-1]);
                }
            Ansx+=Kruskal();
            Ans=Min(Ans,Ansx);
        }
        printf("%lld\n",Ans);
        if(T) printf("\n");
    }
    return 0;
}