UVA 1151二进制枚举子集 + 最小生成树

题意：平面上有n个点(1<=N<=1000)，你的任务是让所有n个点连通，为此，你可以新建一些边，费用等于两个端点的欧几里得距离的平方。另外还有q(0<=q<=8)个套餐（数量小，可枚举），可以购买，如果你购买了第i个套餐，该套餐中的所有结点将变得相互连通，第i个套餐的花费为ci。

分析：按照刘汝佳的思路做的。首先求一次本身的最小生成树值，然后枚举购买的套餐（二进制枚举），每次购买了之后，将其权值设为0，并且加进最小生成树。

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define ll long long
 using namespace std;
 const int maxn=;
 int x[maxn],y[maxn],p[maxn];
 #define repu(i, a, b) for(int i = (a); i < (b); i++)
 int bb[];
 int tran(int h)
 {
     int i = ;
     while(h)
     {
         bb[i] = h%;
         i++;
         h /= ;
     }
     return i;
 }
 int Find(int x)
 {
     return p[x]==x?x:p[x]=Find(p[x]);
 }
 struct edge
 {
     int u,v,w;
     bool operator<(const edge&a)
     const
     {
         return w<a.w;
     }
 } _e[maxn*maxn],e[maxn];
 int q[][maxn],c[],t[];
 int n,m,r,cnt;
 void init()
 {
     m=cnt=;
 }
 ll kruscal()
 {
     ll ret=;
     for(int i=; i<n; i++)///一共就只考虑n-1条边
     {
         int x=Find(e[i].u),y=Find(e[i].v);
         if(x!=y)
         {
             ret += e[i].w;
             p[x]=y;
         }
     }
     return ret;
 }
 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         init();
         scanf("%d%d",&n,&r);
         repu(i,,r)
         {
             scanf("%d%d",&t[i],&c[i]);
             repu(j,,t[i]+)
             scanf("%d",&q[i][j]);
         }
         repu(i,,n+)
         scanf("%d%d",&x[i],&y[i]),p[i]=i;
         repu(i,,n)
         repu(j,i+,n+)
         _e[++m]=(edge)
         {
             i,j,(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])
         };
         sort(_e+,_e+m+);
         ll ans=;
         repu(i,,m+)
         {
             int x = Find(_e[i].u),y = Find(_e[i].v);
             if(x != y)
             {
                 ///自己没能想到的方案：
                 ///存下最小生成树的边,而且每次求最小生成树的时候就直接用这些边求
                 e[++cnt]=_e[i];
                 ans+=_e[i].w;
                 p[x]=y;
             }
         }///本身的最小生成树
         repu(b,,<<r)
         {
             int tt = tran(b);
             ll ansx = ;
             repu(i,,n+) p[i] = i;
             repu(i,,tt)
             {
                 if(bb[i])///如果是1就选该套餐
                 {
                     ansx += c[i];///枚举加哪个套餐，二进制枚举，最多就是2^8个
                     repu(j,,t[i]+)
                     {
                         int xx = Find(q[i][j-]);
                         int yy = Find(q[i][j]);
                         p[xx] = yy;///加入最小生成树
                     }
                 }
             }
             ansx += kruscal();
             ans=min(ans,ansx);
         }
         printf("%lld\n",ans);
         if(T) printf("\n");
     }
     return ;
 }