K - Constructing Roads (最小生成树&prim)

思路:板子题改了一点,给出所有结点的距离和若干个已经建好的道路(即花费为0),求最小生成树。显然prim解决。

AC代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue> 
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e2+5,inf=0x3f3f3f3f;
int a[N][N],n; 
int prim(){
	int vis[N]={},dis[N]={},ans=0;
	for(int i=1;i<=n;i++) dis[i]=a[1][i];
	vis[1]=1;
	for(int i=2;i<=n;i++){
		int mn=inf,p=0;
		for(int j=1;j<=n;j++)
			if(!vis[j]&&dis[j]<mn)
				mn=dis[j],p=j;
		if(mn==inf) break;
		ans+=mn,vis[p]=1;
		for(int j=1;j<=n;j++)
			if(!vis[j]&&a[p][j]<dis[j])
				dis[j]=a[p][j];
	}
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++) scanf("%d",&a[i][j]);
	int q;
	scanf("%d",&q);
	while(q--){
		int u,v;
		scanf("%d%d",&u,&v);
		a[u][v]=a[v][u]=0;
	}
	printf("%d\n",prim());
	return 0;
}