洛谷 P1194 买礼物

题目来源于：洛谷

 

题目本质：图论+生成树

代码如下：

#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;
}