UVA 1151 Buy Or Build 最小生成树+子集枚举

{0,1,2，...,n-1}的集合能组成的数有0-2^n-1，其中，数值为s的组合（只有一个）有

void print_subset(int n,int s)
{
    for(int i=0;i<n;++i)
        if(s&(1<<i)) printf("%d ",i);
    printf("\n");
}

s&(1<<i)相当于求s与{i}的交集，亦即s的第i位是否为1

枚举所有子集

for(int i=0;i<(1<<n);++i)
    print_subset(n,i);

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
LL oo=1e16;
struct edge
{
    int u,v,w;
    bool operator < (const edge & rhs) const
    {
        return w<rhs.w;
    }
}g[1000010],a[1010];        //a数组存没选套餐时的最小生成树的边,g存的是所有边
int m,n,q,fa[1010],num[10],w[10],f[10][1010],tot,xx[1010],yy[1010];
LL ans;
LL dis(int i,int j)
{
    return (LL)((xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j]));
}
int find(int x)
{
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
void init()
{
    int i,j;
    ans=oo;
    scanf("%d%d",&n,&q);
    for (i=1;i<=q;i++)
    {
        scanf("%d%d",&num[i],&w[i]);
        for (j=1;j<=num[i];j++)
          scanf("%d",&f[i][j]);
    }
    for (i=1;i<=n;i++)
      scanf("%d%d",&xx[i],&yy[i]);
    tot=0;
    for (i=1;i<=n;i++)
      for (j=i+1;j<=n;j++)
        g[++tot]=(edge){i,j,dis(i,j)};
    sort(g+1,g+tot+1);
}
void pre()
{
    int i,cnt=0;
    for (i=1;i<=n;i++) fa[i]=i;
    for (i=1;cnt<n-1;i++)
      if (find(g[i].u)!=find(g[i].v))
      {
        fa[fa[g[i].u]]=fa[g[i].v];
        a[++cnt]=g[i];
      }
}
void solve(int k)               //值为k时候选择
{
    LL ret=0;
    int i,j,u,v,cnt=0;
    for (i=1;i<=n;i++)
      fa[i]=i;
    for (i=1;i<=q;i++)
      if (k&(1<<(i-1)))
      {
        ret+=w[i];
        for (j=1;j<num[i];j++)
        {
            u=find(f[i][j]);
            v=find(f[i][j+1]);
            if (u!=v)
            {
                fa[u]=v;
                cnt++;
            }
        }
      }
    for (i=1;cnt<n-1;i++)
    {
        int x=find(a[i].u),y=find(a[i].v);
        if (x!=y)
        {
            ret+=a[i].w;
            cnt++;
            fa[x]=y;
        }
    }

    ans=min(ans,ret);
}
int main()
{
    int i,T;
    scanf("%d",&T);
    while (T--)
    {
        init();
        pre();
        for (i=0;i<(1<<q);i++)
          solve(i);
        printf("%lld\n",ans);
        if (T) printf("\n");
    }
}