Kruskal+二进制枚举 POJ 2784 Buy or Build

题目：https://odzkskevi.qnssl.com/95fc8c37ba05359d81df6779e817c236?v=1501937229

因为q很小，所以直接枚举所有的方案就行。
首先求不用方案时的最小生成树，然后每次枚举方案时只需考虑初次最小生成树中的边就可以，如果每次都用所有边会tle。
感觉自己写的又长又蠢。。。
二进制枚举

for(int i=0;i<(1<<q);i++)
    {
          if(i&(1<<j))
            {}
    }

#include <iostream>
#include <cstring>
#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int INF=0x3f3f3f3f;
const int maxn=1000004/2;
int n,q,cnt;
struct Edge
{
    int u,v,len;
}edge[maxn];
struct Way
{
    int num,cost,citys[1005];
}ways[10];
int r[maxn],p[1005];
pii City[1005];
vector<Edge>minrode;
int dis(pii a,pii b)
{
    return (a.first-b.first)*(a.first-b.first)+(a.second-b.second)*(a.second-b.second);
}
int cmp(int a,int b)
{
    return edge[a].len<edge[b].len;
}
void init()
{
    for(int i=0;i<=n;i++)
        p[i]=i;
}
int find(int x)
{
    return p[x]==x?x:p[x]=find(p[x]);
}
int kruskal()
{
    int ans=0;
    for(int i=0;i<cnt;i++)  r[i]=i;
    sort(r,r+cnt,cmp);
    minrode.clear();
    for(int i=0;i<cnt;i++)
    {
        int e=r[i],x=find(edge[e].u),y=find(edge[e].v);
        if(x!=y)
        {
            ans+=edge[e].len;
            p[x]=y;
            minrode.push_back(edge[e]);
        }
    }
    return ans;
}
int __kruskal()
{
    int ans=0;
    //for(int i=0;i<cnt;i++)  r[i]=i;
    //sort(r,r+cnt,cmp);
    for(int i=0;i<minrode.size();i++)
    {
        int x=find(minrode[i].u),y=find(minrode[i].v);
        if(x!=y)
        {
            ans+=minrode[i].len;
            p[x]=y;
        }
    }
    return ans;
}
void solve()
{
    init();
    int ans=kruskal();
    for(int i=0;i<(1<<q);i++)
    {
        init();
        int tol=0;
        for(int j=0;j<q;j++)
        {
            if(i&(1<<j))
            {
                tol+=ways[j].cost;
                for(int k=1;k<ways[j].num;k++)
                {
                    int x=find(ways[j].citys[0]),y=find(ways[j].citys[k]);
                    if(x!=y)
                    {
                        p[y]=x;
                    }
                }
            }
        }
        ans=min(ans,tol+__kruskal());
    }
    cout<<ans<<endl;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>q;
        for(int i=0;i<q;i++)
        {
            cin>>ways[i].num>>ways[i].cost;
            for(int j=0;j<ways[i].num;j++)
                cin>>ways[i].citys[j];
        }
        int x,y;
        for(int i=1;i<=n;i++)
        {
            cin>>x>>y;
            City[i]=make_pair(x,y);
        }
        cnt=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
        {
            edge[cnt].u=i;
            edge[cnt].v=j;
            edge[cnt++].len=dis(City[i],City[j]);
        }
        solve();
        if(t) cout<<endl;
    }
    return 0;
}