【L - 还是畅通工程】

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

思路:

  • 最小生成树裸题,稠密图,比较一下 Prim & Kruskal
    1. Prim:不需要离线做存储边,使用邻接矩阵存储边权即可,用到 Q(NODE)、dis[]、vis[] 。注意初始化、对角线、对称性、单边性
    2. Kruskal:记住是并查集就很好写。

代码:

  • Prim核心:只需要询问新加入点的直连边即可
for(int i=1;i<=N;i++)
	if(!vis[i] && dis[i] > mp[id][i]){
		dis[i] = mp[id][i];
		Q.push(NODE(i , dis[i]));
	}
  • Prim:1045ms 1492kB
//1045ms		1492kB


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

using namespace std;

const int maxn = 105;

int N,M;
int ans;
int dis[maxn];
bool vis[maxn];
int mp[maxn][maxn];
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(){
	ans = 0;
	memset(dis , INF , sizeof(dis));
	memset(vis , 0 , sizeof(vis));
	memset(mp , INF , sizeof(mp));
	for(int i=0;i<maxn;i++)
		mp[i][i] = 0;
	return ;
}

void PRIM(){
	Q.push(NODE(1,0)) ; dis[1] = 0;
	while(Q.size()){
		NODE cur = Q.top() ; Q.pop();
		int id = cur.id ;
		if(vis[id])
			continue;
		vis[id] = true;
		ans += dis[id];
		//接下来的部分是核心,只需要询问新加入点的直连边即可
		for(int i=1;i<=N;i++)
			if(!vis[i] && dis[i] > mp[id][i]){
				dis[i] = mp[id][i];
				Q.push(NODE(i , dis[i]));
			}
	}
	return ;
}

int main(){
	while(cin>>N && N){
		INIT();
		M = N*(N-1)/2;
		while(M--){
			int l,r,w;
			cin>>l>>r>>w;
			mp[l][r] = mp[r][l] = min(mp[l][r] , w);
		}
		PRIM();
		cout<<ans<<endl;
	}
	return 0;
}
  • Kruskal:1014ms 1496kB
//1014ms		1496kB


#include <iostream>
#include <algorithm> 
#include <cstring>

using namespace std;

const int maxn = 105;

int N,M;
int ans;
int par[maxn];
struct EDGE{
	int l,r,w;
	friend bool operator < (EDGE a , EDGE b)
	{
		return a.w < b.w;
	}
}edge[maxn*maxn/2];
void ADDEDGE(int i,int l,int r,int w){
	edge[i].l = l;
	edge[i].r = r;
	edge[i].w = w;
	return ;
}

void INIT(){
	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){
	int parl = FIND(l);
	int parr = FIND(r);
	par[parr] = parl;
	return ;
}

void KRUSKAL(){
	for(int i=0;i<M;i++){
		int l = edge[i].l;
		int r = edge[i].r;
		int parl = FIND(l);
		int parr = FIND(r);
		if(parl != parr){
			UNION(l,r);
			ans += edge[i].w;
		}
	}
	return ;
}

int main(){
	while(cin>>N && N){
		INIT();
		M = N*(N-1)/2;
		for(int i=0;i<M;i++){
			int l,r,w;
			cin>>l>>r>>w;
			ADDEDGE(i,l,r,w);
		} 
		sort(edge , edge+M);
		KRUSKAL();
		cout<<ans<<endl;
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值