【QBKbupt】第七章 图 D. 图练习4

#include<bits/stdc++.h>
using namespace std;
const int MAX=0x3fffffff;
int n,a[21][21],v[21],d[21],ans;
int prime();
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) scanf("%d",&a[i][j]);
	printf("%d",prime());
}
int prime()
{
	int t,num;
	d[1]=0;
	v[1]=1;
	for(int i=2;i<=n;i++) d[i]=MAX,v[i]=0;
	for(int i=1;i<=n;i++) if(a[1][i]!=0&&d[i]>a[1][i]) d[i]=a[1][i];
	while(1)
	{
		t=MAX;
		num=0;
		for(int i=1;i<=n;i++) if(v[i]==0&&d[i]<t) t=d[i],num=i;
		if(num==0) break;
		v[num]=1;
		ans+=t;
		for(int j=1;j<=n;j++) if(a[num][j]!=0&&v[j]==0&&d[j]>a[num][j]) d[j]=a[num][j];
	}
	return ans;
}