UVA 1151 Buy or Build（最小生成树+枚举子集）

题目地址：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3592

思路：单纯枚举时间复杂度难以承受。则先对原图求最小生成树得到n-1条边，每次枚举套餐时仅考虑套餐中的边和这n-1边（将套餐中的边权值置为0和这n-1条边求最小生成树即可）。

#include<cstdio>
#include<cmath>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define debu
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=1000+50;
const int maxm=1000000+50;
struct Node
{
    int u,v,w;
    Node(int u=0,int v=0,int w=0):u(u),v(v),w(w) {}
};
struct Point
{
    int x,y;
};
int ans;
int n,tot,m,q;
int fa[maxn];
Point p[maxn];
Node e[maxm];
vector<int> a[10];
vector<Node> Edge;
vector<Node> tmpEdge;
void init()
{
    for(int i=0; i<=n; i++) fa[i]=i;
}
inline int Find(int x)
{
    return fa[x]==x?x:fa[x]=Find(fa[x]);
}
inline double Dist(int i,int j)
{
    return ((p[i].x-p[j].x)*(p[i].x-p[j].x)+(p[i].y-p[j].y)*(p[i].y-p[j].y));
}
int cmp(Node a,Node b)
{
    return a.w<b.w;
}
void build()
{
    init();
    int num=0,tot=0;
    for(int i=0; i<n; i++)
        for(int j=i+1; j<n; j++)
            e[num++]=Node(i,j,Dist(i,j));
    sort(e,e+num,cmp);
    for(int i=0; i<num; i++)
    {
        int x=Find(e[i].u);
        int y=Find(e[i].v);
        if(x!=y)
        {
            fa[x]=y;
            tot++;
            ans+=e[i].w;
            Edge.push_back(Node(e[i].u,e[i].v,e[i].w));
        }
        if(tot==n-1) break;
    }
}
void input()
{
    scanf("%d%d",&n,&q);
    for(int i=0; i<q; i++) a[i].clear();
    for(int i=0; i<q; i++)
    {
        int num,cost;
        scanf("%d%d",&num,&cost);
        a[i].push_back(cost);
        for(int j=0; j<num; j++)
        {
            int x;
            scanf("%d",&x);
            a[i].push_back(x);
        }
    }
    for(int i=0; i<n; i++)
        scanf("%d%d",&p[i].x,&p[i].y);
}
void solve()
{
    for(int s=0; s<(1<<q); s++)
    {
        init();
        tmpEdge=Edge;
        int tot=0,tmp=0;
        for(int i=0; i<q; i++)
        {
            if(s&(1<<i))
            {
                tmp+=a[i][0];
                for(int j=1; j<a[i].size(); j++)
                    for(int k=j+1; k<a[i].size(); k++)
                        tmpEdge.push_back(Node(a[i][j]-1,a[i][k]-1,0));
            }
        }
        sort(tmpEdge.begin(),tmpEdge.end(),cmp);
        for(int i=0; i<tmpEdge.size(); i++)
        {
            int x=Find(tmpEdge[i].u);
            int y=Find(tmpEdge[i].v);
            if(x!=y)
            {
                fa[x]=y,tot++;
                tmp+=tmpEdge[i].w;
                if(tot==n-1)
                {
                    ans=min(ans,tmp);
                    break;
                }
            }
        }
    }
    printf("%d\n",ans);
}
int main()
{
#ifdef debug
    freopen("in.in","r",stdin);
#endif // debug
    int t,cas=0;
    scanf("%d",&t);
    while(t--)
    {
        Edge.clear(),ans=0;
        if(++cas!=1) printf("\n");
        input();
        build();
        solve();
    }
    return 0;
}