Jungle Roads【最小生成树模板题】

题目描述

就是裸的最小生成树

知识点

最小生成树

我的实现

码前思考

考虑到多条边的情况，使用邻接表；
考虑顶点数固定，使用prim~

代码实现

//对于这种图的顶点比较少的图，我们使用prim
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 30;
const int inf = 0x3fffffff;

struct node{
	int v;
	int dis;
	node(){}
	node(int _v,int _dis){
		v=_v;
		dis=_dis;
	}
};

vector<node> adj[maxn];
int n;

int prim(){
	int ans=0;
	bool vis[maxn]={false};
	int d[maxn];
	fill(d,d+maxn,inf);
	d[0]=0;
	for(int i=0;i<n;i++){
		int u=-1,MIN=inf;
		for(int j=0;j<n;j++){
			if(vis[j]==false&&d[j]<MIN){
				u=j;
				MIN=d[j];
			}
		}
		ans+=d[u];
		vis[u]=true;
		
		for(int j=0;j<adj[u].size();j++){
			int v = adj[u][j].v;
			int dis = adj[u][j].dis;
			if(vis[v]==false&&dis<d[v]){
				d[v]=dis;
			}
		}
	}
	return ans; 
}

int main(){
	//freopen("input.txt","r",stdin);
	while(scanf("%d",&n)!=EOF){
		if(n==0){
			return 0;
		} 
		for(int i=0;i<maxn;i++){
			adj[i].clear();
		}
		for(int i=0;i<n-1;i++){
			getchar();//吸收换行？ 
			char tmp;
			scanf("%c ",&tmp);
			int u=i;
			int k;
			scanf("%d ",&k);
			for(int j=0;j<k;j++){
				int v;
				char c;
				int dis;
				scanf("%c %d ",&c,&dis);
				v=c-'A';
				adj[u].push_back(node(v,dis));
				adj[v].push_back(node(u,dis));
			} 
		}
		//运行prim算法
		printf("%d\n",prim()); 
	}	
	return 0;
} 

码后反思

输入卡了好久，scanf读取字符时，换行和空格都会被读取。所以需要注意这一点。