hiho 最小生成樹二(Kruscal算法)

題目:

#1098 : 最小生成樹二·Kruscal算法

時間限制:10000ms
單點時限:1000ms
內存限制:256MB

描述

隨着小Hi擁有城市數目的增加,在之間所使用的Prim算法已經無法繼續使用了——但是幸運的是,經過計算機的分析,小Hi已經篩選出了一些比較適合建造道路的路線,這個數量並沒有特別的大。

所以問題變成了——小Hi現在手上擁有N座城市,且已知其中一些城市間建造道路的費用,小Hi希望知道,最少花費多少就可以使得任意兩座城市都可以通過所建造的道路互相到達(假設有A、B、C三座城市,只需要在AB之間和BC之間建造道路,那麼AC之間也是可以通過這兩條道路連通的)。

提示:積累的好處在於可以可以隨時從自己的知識庫中提取想要的!

輸入

每個測試點(輸入文件)有且僅有一組測試數據。

在一組測試數據中:

第1行爲2個整數N、M,表示小Hi擁有的城市數量和小Hi篩選出路線的條數。

接下來的M行,每行描述一條路線,其中第i行爲3個整數N1_i, N2_i, V_i,分別表示這條路線的兩個端點和在這條路線上建造道路的費用。

對於100%的數據,滿足N<=10^5, M<=10^6,於任意i滿足1<=N1_i, N2_i<=N, N1_i≠N2_i, 1<=V_i<=10^3.

對於100%的數據,滿足一定存在一種方案,使得任意兩座城市都可以互相到達。

輸出

對於每組測試數據,輸出1個整數Ans,表示爲了使任意兩座城市都可以通過所建造的道路互相到達至少需要的建造費用。

樣例輸入
5 29
1 2 674
2 3 249
3 4 672
4 5 933
1 2 788
3 4 147
2 4 504
3 4 38
1 3 65
3 5 6
1 5 865
1 3 590
1 4 682
2 4 227
2 4 636
1 4 312
1 3 143
2 5 158
2 3 516
3 5 102
1 5 605
1 4 99
4 5 224
2 4 198
3 5 894
1 5 845
3 4 7
2 4 14
1 4 185
樣例輸出
92
題解:利用Kruscal算法,先將邊排序,依次找到m-1條最短的邊,且不在找到的集合中,利用並查集

代碼:

#include<iostream>
#include<algorithm>
#define M 1000006
using namespace std;
int fa[M];
struct Node{
	int l;
	int r;
	int length;
}s[M];
bool cmp(Node x,Node y)
{
  return x.length<y.length;	
} 
//找父節點 
int find(int x)
{
	if(x==fa[x])
	{
		return x;
	}
	else
	return fa[x]=find(fa[x]);
}
int main()
{
	int m,n,i,j;
	cin>>m>>n;
	for(i=0;i<n;i++)
	cin>>s[i].l>>s[i].r>>s[i].length;
	sort(s,s+n,cmp);
	
	for(i=1;i<=m;i++)
	fa[i]=i;
	
	int sum=0;
	int k=0;
	int x,y;
	
	for(i=0;i<m-1;i++)
	{
		for(j=k;j<n;j++)
		{
			x=find(s[j].l);
			y=find(s[j].r);
			if(x==y)
			{
				continue;
			}
			
			fa[x]=y;
			sum=sum+s[j].length;
			k=j+1;
			break;
		}
	}
	cout<<sum<<endl;
	return 0;
	
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章