HDU - 1102 Constructing Roads

https://vjudge.net/problem/HDU-1102

注意更新的时候要双向更新为0

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=120;
int n,m;
int g[N][N],dis[N];
bool bk[N];
int prim()
{
	memset(bk,false,sizeof bk);
	memset(dis,0x3f,sizeof dis);
	int ans=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!bk[j] && (t==-1 || dis[t]>dis[j]))
				t=j;
		}
		bk[t]=1;
		if(i) ans+=dis[t];
		
		for(int j=1;j<=n;j++)
			if(dis[j]>g[t][j])
				dis[j]=g[t][j];
	}
	return ans;
}
int main()
{
	while(~scanf("%d",&n))
	{
		memset(g,0x3f,sizeof g);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
				scanf("%d",&g[i][j]);
		}
		
		int q;
		scanf("%d",&q);
		int a,b;
		for(int i=0;i<q;i++)
		{
			scanf("%d %d",&a,&b);
			g[a][b]=0;
			g[b][a]=0;
		}
		
		printf("%d\n",prim());
	}
	return 0;
}