博客 UVA 1151 Buy or Bulid 生成树 图论

题目大意：给定一个图和图的一些子网，求连接图的最小花费；

连接图--- kruskal 并查集

给定有子网---运用二进制枚举子网

具体思路： 先求一次MST，得到需要的边。枚举子网，通过并查集将子网中的点加入，再去需要的边中找剩余所需要的cnt条边

选择最小的ans；

源代码：

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=1000 +10;
const int maxq=9;
int xx[maxn],yy[maxn],cost[maxq]
int pre[maxn];                         //并查集数组
vector<int >subn[maxq];                //子网载体
struct Edge{
    int u,v,d;
    Edge(int u,int v,int d):u(u),v(v),d(d){}  //方便后序的插入操作
    bool operator <(const Edge &a)            //用于排序
    const  {return d<a.d;}
};
int find(int lin){                     //并查集查找函数（路径优化）
    int i=lin;
    while(pre[i]!=i)i=pre[i];
    int j=lin,k;
    while(j!=pre[j]){
        k=pre[j];
        pre[j]=i;
        j=k;
    }
    return i;
}
int MST(int cnt,const vector<Edge> & e,vector<Edge>& used){ //最小生成树 重点为cnt的值
    if(cnt==1)return 0;
    int ans=0;
    int m=e.size();
    used.clear();
    for(int i=0;i<m;i++){
        int x=find(e[i].u),y=find(e[i].v);
        int d=e[i].d;
        if(x!=y){
            pre[x]=y;
            ans+=d;
            used.push_back(e[i]);    //装入生成MST所需要用到的边，用于后续的子集生成模块
            if(--cnt==1)break;
        }
    }
    return ans;
}

int main(){
    int T,n,q;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&q);
        for(int i=0;i<q;i++){
            int sum,c,temp;
            scanf("%d%d",&sum,&c);
            cost[i]=c;
            subn[i].clear();
            while(sum--){
                scanf("%d",&temp);
                subn[i].push_back(temp-1);
            }
        }
        for(int i=0;i<n;i++)scanf("%d%d",&xx[i],&yy[i]);
        
        vector<Edge> e, need;
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                int c=(xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j]);
                e.push_back(Edge(i,j,c));
            }
        }
        
        for(int i=0;i<n;i++)pre[i]=i;
        sort(e.begin(),e.end());
        int ans=MST(n,e,need);
        
        vector<Edge>sun;
        for(int mask=0;mask< (1<<q);mask++){     //子集生成模块
            int cnt=n,c=0;
            for(int i=0;i<n;i++)pre[i]=i;        //子网中加入的点与后序的MST都是这一个集合，子网已连接的在后续MST中不会再查找
            for(int i=0;i<q;i++)if(mask& (1<<i)){
                c+=cost[i];
                int ss=subn[i].size();
                for(int j=0;j<ss;j++){
                    int u=find(subn[i][j]),v=find(subn[i][0]);   //将子网中的点加入集合。
                    if(u!=v){pre[u]=v;--cnt;}
                }
            }
            ans=min(ans, c+ MST(cnt,need,sun)); // 二次MST
        }
        printf("%d\n",ans);
        if(T)printf("\n");
    }
    return 0;
}

转载于:https://www.cnblogs.com/zhizhaozhuo/p/9594230.html