最小生成树prim之挖井

题目链接：

 

 

#include <iostream>
using namespace std;
int G[310][310];
int n,u;
double key[310];
bool M[310];
int ans=0;
void prim()
{
	//int u;定义的有问题
	double  min;
	for(int i=1;i<=n+1;i++)
	{
		min=99999999;
		for(int j=1;j<=n+1;j++)
			if(M[j]&&key[j]<min)
			{
				min=key[i];
				u=j;
			}
		ans+=min;
		M[u]=false;
		for(int j=1;j<=n+1;j++)
			if(M[j]&&G[u][j]!=-1&&key[j]>G[u][j])
				key[j]=G[u][j];
	}
}
int main()
{
	cin>>n;
	int a;
	for(int i=1;i<=n;i++)
	{
		cin>>a;
		G[n+1][i]=G[i][n+1]=a;
	}
	G[n+1][n+1]=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			cin>>G[i][j];

	for(int i=1;i<=n;i++)
	{
		M[i]=true;
		key[i]=99999999;
	}
	key[n+1]=0;
	prim();
	cout<<ans;
	system("pause");
	return 0;
}