zoj_2326 Tangled in Cables

伪神

于 2013-05-23 19:31:16 发布

阅读量547

点赞数

CC 4.0 BY-SA版权

分类专栏： zoj 文章标签：并查集 kruskal

本文链接：https://blog.youkuaiyun.com/LOVE_ACM_LOVE_MM/article/details/8966866

zoj 专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种使用Kruskal算法解决最小生成树问题的方法，通过C++代码实现了基于顶点并查集的高效求解过程。该算法适用于计算网络中各节点间的最短连接路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=1381

//C++代码
#include<iostream>
#include<iomanip>
#include<string>
#include<queue>
#include<map>
using namespace std;

struct Edge{
	int u,v;
	double w;
	bool operator <(const Edge a) const{
		return w>a.w;
	}
}edge;

int n,s[1005];
double length;
priority_queue<Edge> pq;

void Union(int x,int y){
	s[y]=x;
}

int Find(int x){
	if(s[x]<0) return x;
	return s[x]=Find(s[x]);
}

void Kruskal(){
	int num=0,i;
	double sum=0;
	for(i=1;i<=n;i++) s[i]=-1;
	while(!pq.empty() && num<n-1){
		edge=pq.top();
		pq.pop();
		if(Find(edge.u)!=Find(edge.v)){
			Union(Find(edge.u),Find(edge.v));
			sum+=edge.w;
			num++;
		}
	}
	if(sum>length) cout<<"Not enough cable"<<endl;
	else cout<<"Need "<<fixed<<setprecision(1)<<sum<<" miles of cable"<<endl;
}

int main(){
	int m,i;
	string s,a,b;
	map<string,int> p;
	cin>>length>>n;
	for(i=1;i<=n;i++){
		cin>>s;
		p.insert(make_pair(s,i));
	}
	cin>>m;
	for(i=1;i<=m;i++){
		cin>>a>>b>>edge.w;
		edge.u=p.find(a)->second;
		edge.v=p.find(b)->second;
		pq.push(edge);
	}
	Kruskal();
	return 0;
}