csu 1116 Kingdoms 最小生成树 二进制枚举

来源 湖南省第八届大学生计算机程序设计竞赛
题目大题
有n个城市，m条边，选取一些边使得某些个城市连通（必须包含1号顶点），从而使得这些城市拥有的人口最多，并且修建他们之间的道路的费用不超过k
解题思路
因为才16个城市，100条边，3000ms，二进制枚举可能的城市集合，并用最小生成树求出总费用，取最大的人口就可了
还可以用状态压缩dp，以后再更新

代码很挫 轻喷

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

int p[20];

void init(int n)
{
    for(int i=1;i<=n;i++)
        p[i]=i;
}

int findx(int x)
{
    if(p[x]==x)
        return x;
    else
        return p[x]=findx(p[x]);
}

void unite(int x,int y)
{
    x=findx(x);
    y=findx(y);
    if(x!=y)
        p[x]=y;
}

bool same(int x,int y)
{
    return findx(x)==findx(y);
}

vector<pair<int,int> >G[20];
int popu[20];

struct node
{
    int u,v,cost;
    node(int x,int y,int z)
    {
        u=x;v=y;cost=z;
    }
};

bool cmp(node a,node b)
{
    return a.cost<b.cost;
}
int n,m,k;

//判断由二进制枚举出来的城市是否满足费用小于k，并返回population
int judge(int s)
{
    //初始化
    init(16);
    vector<node> g;//存边
    vector<int> E;//存点
    int cnt=0;
    int vis[20];//判断下标所代表的城市是否存在
    memset(vis,0,sizeof(vis));

    for(int c=0;c<n;c++)
    {
        if(s & (1<<c))//第c位是否是1
        {
            E.push_back(c+1);
            cnt++;
            vis[c+1]=1;
        }
    }
    if(!vis[1]) return 0;//1作为中心城市必须包含
    for(int i=0;i<cnt;i++)
    {
        int e=G[E[i]].size();
        for(int j=0;j<e;j++)
        {
            if(vis[G[E[i]][j].first])
                {
                    g.push_back(node(E[i],G[E[i]][j].first,G[E[i]][j].second));

                }
        }
    }
    //最小生成树
    sort(g.begin(),g.end(),cmp);
    int res=0;
    for(int i=0;i<g.size();i++)
    {
        node e=g[i];
        if(!same(e.u,e.v))
        {
            unite(e.u,e.v);
            res+=e.cost;
        }
    }
    int t=0;
    int ans=0;
    for(int i=0;i<cnt;i++)
    {
       if(findx(E[i])==E[i])
         t++;
       ans+=popu[E[i]];
    }

    if(t!=1) return 0;//无法组成一个集合
    if(res>k) return 0;//费用大于k
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<=n;i++)
            G[i].clear();
        memset(popu,0,sizeof(popu));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&popu[i]);
        }
        for(int i=0;i<m;i++)
        {
            int u,v,cost;
            scanf("%d%d%d",&u,&v,&cost);
            //用邻接表存边
            G[u].push_back(make_pair(v,cost));
            G[v].push_back(make_pair(u,cost));
        }
        int mx=0;
        for(int i=1;i<(1<<n);i++)
        {
            mx=max(mx,judge(i));//取最大的人口
        }
        printf("%d\n",mx);

    }
    return 0;

}