题目大意:给定一个图和图的一些子网,求连接图的最小花费;
连接图--- 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;
}