【A - Jungle Roads】

80 篇文章 0 订阅
80 篇文章 0 订阅

思路:

  • 最小生成树板题。
  • Kruskal(并查集):存储所有sort,遍历一遍边即可找出所有应采用的边,因此也可以使用优先队列。不需要 vis[],并查集足矣。
  • Prim(优先队列):类似 Dijkstra,以态存优先队列,相当于全图最短路,有所不同的是,最短路松弛为 dis[v] = dis[u] + w;最小生成树松弛为 dis[v] = w 。利用 vis[]。可以用邻接表或邻接矩阵。本代码都没用,直接存边了。

代码:

  • Kruskal:0ms 684kB
//0ms		684kB


#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 105;

int N;
int ans;
int par[maxn];
struct EDGE{
	int u,v,w;
	friend bool operator < (EDGE a , EDGE b){
		return a.w < b.w;
	}
}edge[maxn];int cnt = 0;//from 1 .. cnt
void ADDEDGE(int u,int v,int w){
	cnt++;
	edge[cnt].u = u;
	edge[cnt].v = v;
	edge[cnt].w = w;
	return ;
}

void INIT(){
	cnt = 0;
	ans = 0;
	memset(par , -1 , sizeof(par));
	return ;
}

int FIND(int i){
	return par[i] == -1 ? i : par[i] = FIND(par[i]) ;
}

void UNION(int l,int r){
	par[r] = l;
	return ;
}

void KRUSKAL(){
	int k=1;
	while(k<N){
		for(int i=1;i<=cnt;i++){
			int l = edge[i].u;
			int r = edge[i].v;
			int parl = FIND(l);
			int parr = FIND(r);
			if(parl != parr){
				UNION(parl,parr);
				ans += edge[i].w;
				k++;
			}
		}
	}
	return ;
}

int main(){
	while(cin>>N && N){
		INIT();
		for(int i=1;i<N;i++){
			int k;
			char op;
			cin>>op>>k;
			int u = op-'A';
			while(k--){
				int w;
				cin>>op>>w;
				ADDEDGE(u , int(op-'A') , w);
			}
		}
		sort(edge+1 , edge+cnt+1);
		KRUSKAL();
		cout<<ans<<endl;
	}
	return 0;
}
  • Prim:0ms 676kB
//0ms		676kB


#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;

const int maxn = 105;

int N;
int ans;
int dis[maxn];
bool vis[maxn];
struct EDGE{
	int u,v,w;
	friend bool operator < (EDGE a , EDGE b){
		return a.w < b.w;
	}
}edge[maxn];int cnt;//from 1 .. cnt
void ADDEDGE(int u,int v,int w){
	cnt++;
	edge[cnt].u = u;
	edge[cnt].v = v;
	edge[cnt].w = w;
	return ;
}
struct NODE{
	int id;
	int dis;
	friend bool operator > (NODE a , NODE b)
	{
		return a.dis > b.dis;
	}
	NODE(int id,int dis) : id(id) , dis(dis) {} ;
};
priority_queue <NODE , vector<NODE> , greater<NODE> > Q;

void INIT(){
	cnt = 0;
	ans = 0;
	memset(dis , INF , sizeof(dis));
	memset(vis , 0 , sizeof(vis));
	return ;
}

void PRIM(){
	dis[0] = 0;
	Q.push(NODE(0 , 0));
	while(Q.size()){
		NODE cur = Q.top() ; Q.pop();
		int id = cur.id;
		if(!vis[id]){
			vis[id] = true;
			ans += dis[id];
			for(int i=1;i<=cnt;i++){
				//小心写代码。 
				int u = edge[i].u;
				int v = edge[i].v;
				int w = edge[i].w;
				if(u == id && !vis[v] && dis[v] > w){
					dis[v] = w;
					Q.push(NODE(v , dis[v])); 
				}
				if(v == id && !vis[u] && dis[u] > w){
					dis[u] = w;
					Q.push(NODE(u , dis[u]));
				}
			}
		}
	}
	return ;
}

int main(){
	while(cin>>N && N){
		INIT();
		for(int i=1;i<N;i++){
			int k;
			char op;
			cin>>op>>k;
			int u = op-'A';
			while(k--){
				int w;
				cin>>op>>w;
				ADDEDGE(u , int(op-'A') , w);
			}
		}
		PRIM();
		cout<<ans<<endl;
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值