SDUT 2144-最小生成樹

圖結構練習——最小生成樹

Time Limit: 1000MSMemory Limit: 65536KB

Problem Description

 有n個城市，其中有些城市之間可以修建公路，修建不同的公路費用是不同的。現在我們想知道，最少花多少錢修公路可以將所有的城市連在一起，使在任意一城市出發，可以到達其他任意的城市。

 

Input

 輸入包含多組數據，格式如下。

第一行包括兩個整數n m，代表城市個數和可以修建的公路個數。(n <= 100， m <=10000)

剩下m行每行3個正整數a b c，代表城市a 和城市b之間可以修建一條公路，代價爲c。

 

Output

 每組輸出佔一行，僅輸出最小花費。

Example Input

3 2
1 2 1
1 3 1
1 0

Example Output

2
0

就是普里姆算法，沒什麼特別的，另外還有個坑

#include<iostream>
#include<string.h>
using namespace std;
#define max 100000
int ma[101][101],lowc[101],vis[101],n,m;
int prim()
{
	int mi,flag,sum=0;
	for(int i=0;i<=n;i++) lowc[i]=ma[1][i];//這裏從1開始遍歷（從哪開始都可以）
	vis[1]=1;
	for(int i=1;i<n;i++)
	{
		mi=max;
		flag=0;
		for(int j=0;j<=n;j++)
		{
			if(!vis[j]&&lowc[j]<mi)//找出代價最小的
			{
				mi=lowc[j];
				flag=j;
			}
		}
		sum+=mi;//代價累加
		vis[flag]=1;
		for(int j=0;j<=n;j++)//更新lowc
		{
			if(!vis[j]&&lowc[j]>ma[flag][j])
			{
				lowc[j]=ma[flag][j];
			}
		}
	}
	return sum;
}
int main()
{
	int a,b,q;
	while(cin>>n>>m)
	{
		for(int i=0;i<=n;i++)
		{
			for(int j=0;j<=n;j++)
			{
				if(i==j) ma[i][j]=0;
				else ma[i][j]=max;
			}
		}
		for(int i=0;i<m;i++)
		{
			cin>>a>>b>>q;
			if(q<ma[a][b])//坑在這裏！！！！！我本來以爲這步沒必要，但是提交後WA！！！加上才AC了，真的氣！！！
			{
				ma[a][b]=ma[b][a]=q;
			}
		}
		memset(vis,0,sizeof(vis));
		cout<<prim()<<endl;
	}
	return 0;
}