UVa 1151 Buy or Build【最小生成树】

题意：给出n个点的坐标，现在需要让这n个点连通，可以直接在点与点之间连边，花费为两点之间欧几里得距离的平方，也可以选购套餐，套餐中所含的点是相互连通的 问最少的花费

首先想kruskal算法中，被加入的边已经是最优的了，所以当选择完套餐后，之前被丢弃的边也不会再进入最小生成树

然后就可以先求一次原图的最小生成树，保存下进入最小生成树的n-1条边

再枚举选择的套餐的情况，再求最小生成树，这里用的二进制法枚举 最后维护一个最小值就可以了

思路虽然看懂了，可是代码根本就写不出来，看着标程写的，最后还是改了那么久-- sad----------

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include <cmath>
 #include<stack>
 #include<vector>
 #include<map>
 #include<set>
 #include<queue>
 #include<algorithm>
 using namespace std;

 #define foreach(i,c) for (__typeof(c.begin()) i = c.begin(); i != c.end(); ++i)

 typedef long long LL;
 const int INF = (<<)-;
 const int mod=;
 const int maxn=;
 const int maxq=;

 int n;
 int x[maxn],y[maxn],cost[maxn];
 vector<int> subn[maxn];

 int p[maxn];
 int find(int x) {return p[x]!=x? p[x]=find(p[x]):x;}

 struct edge{
     int u,v,d;
     edge(int u,int v,int d):u(u),v(v),d(d) {}
     bool operator <(const edge& rhs) const{
     return d<rhs.d;}
 };

 int mst(int cnt,const vector<edge>& e,vector<edge>& used){
     if(cnt==) return ;
     int m=e.size();
     int ans=;
     used.clear();
     for(int i=;i<m;i++){
         int u=find(e[i].u),v=find(e[i].v);
         int d=e[i].d;
         if(u!=v){
             p[u]=v;
             ans+=d;
             used.push_back(e[i]);
             if(--cnt==) break;
         }
     }
     return ans;
 }

 int main(){
 //    freopen("in.txt","r",stdin);
 //    freopen("outttttttt.txt","w",stdout);
     int T,q;
     scanf("%d",&T);
     while(T--){
         scanf("%d%d",&n,&q);
         for(int i=;i<q;i++){
             int cnt;
             scanf("%d %d",&cnt,&cost[i]);
             subn[i].clear();
             while(cnt--){
                 int u;
                 scanf("%d",&u);
                 subn[i].push_back(u-);
             }
         }

         for(int i=;i<n;i++) scanf("%d %d",&x[i],&y[i]);

         vector<edge> e,need;

         for(int i=;i<n;i++)
          for(int j=i+;j<n;j++){
              int c=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
              e.push_back(edge(i,j,c));
          }

         for(int i=;i<n;i++) p[i]=i;
         sort(e.begin(),e.end());

         int ans=mst(n,e,need);

         for(int mask=;mask<(<<q);mask++){

             for(int i=;i<n;i++) p[i]=i;
             int cnt=n,c=;

             for(int i=;i<q;i++) if(mask & (<<i)){
                 c+=cost[i];
                 for(int j=;j<subn[i].size();j++){
                     int u=find(subn[i][j]),v=find(subn[i][]);
                     if(u!=v){p[u]=v;cnt--;}
                 }
             }
             vector<edge> dummy;
             ans=min(ans,c+mst(cnt,need,dummy));
         }
         printf("%d\n",ans);
         if(T) printf("\n");
     }
     return ;
 }