POJ-2421 Constructing Roads

题目链接：http://acm.hust.edu.cn/vjudge/contest/66965#problem/D

题目大意：给你N个村庄之间的距离，再给你M个村庄已修好路，求将这N个村庄连接的最小距离和。

解题思路：最小生成树。直接读入邻接矩阵，在修改其中一些点为0，用Prim求即可。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
int n,m;
int G[102][102];
int ans;
bool vis[102];
void Prim(){
	int s=1;vis[1]=1;
	for(int h=1;h<n;h++){
		int temp=1<<30;int k;
		for(int i=2;i<=n;i++){
			if(!vis[i]&&G[s][i]<temp){
				temp=G[s][i];
				k=i;
			}
		}
		vis[k]=1;
		ans+=temp;
		for(int i=2;i<=n;i++)
			if(!vis[i]&&G[s][i]>G[k][i]) G[s][i]=G[k][i];  
	}
	
	
	
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			scanf("%d",&G[i][j]);
		} 
	scanf("%d",&m);	
	for(int i=1;i<=m;i++){
		int a,b;
		scanf("%d%d",&a,&b);
		G[a][b]=G[b][a]=0; 
	} 
	Prim();
	cout<<ans<<endl;
	return 0;
}