题目来源于:洛谷
题目本质:图论+生成树
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int N=10005;
const int M=250005;
int n,m;
int cnt,flag,px,py;
int ans=0; //ans表示最小花费的钱数
int f[N];
struct Edge{
int x,y,z;
}a[M];
int find(int x){
if (x==f[x]){
return x;
}
return f[x]=find(f[x]);
}
bool cmp(Edge a,Edge b){
return a.z<b.z;
}
int main(){
scanf("%d %d",&n,&m); //n,m分别指题目中的A,B;
for(int i=0;i<=m;i++){
f[i]=i;
}
for(int i=1;i<=m;i++){
cnt++;
a[cnt].x=0;
a[cnt].y=i;
a[cnt].z=n;
}
for(int i=1;i<=m;i++)
for(int j=1;j<=m;j++){
scanf("%d",&flag);
if(flag){
cnt++;
a[cnt].x=i;
a[cnt].y=j;
a[cnt].z=flag;
}
}
sort(a+1,a+cnt+1,cmp);
for (int i=1;i<=cnt;i++){
px=find(a[i].x);
py=find(a[i].y);
if (px==py){
continue;
}
f[px]=py;
ans+=a[i].z;
}
printf("%d\n",ans);
return 0;
}