还是畅通工程

roblem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

  
  

   
   3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
  
  

 

Sample Output

  
  

   
   3
5


   
   

    
    

     
     Hint
    
    
Hint
   
   
   
    
Huge input, scanf is recommended.
  
  

普利姆+堆优化：

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

using namespace std;

struct Node
{
	int to;
	int dis;
	Node(int a,int b)
	{
		to = a;
		dis = b;
	}
	bool operator< (Node t) const
	{
		return dis > t.dis;
	}
};

int map[103][103];
bool visited[103]; 
int ans;

void prim(int n,int x)
{
	memset(visited,false,sizeof(visited));
	priority_queue<Node> q;
	q.push(Node(1,0));
	int cnt = 0;
	ans = 0;
	while(!q.empty())
	{
		Node t = q.top();
		q.pop();
		if(visited[t.to])
		{
			continue;
		}
		visited[t.to] = true;
		ans += t.dis;
		cnt++;
		for(int i = 1;i <= n;i++)
		{
			if(!visited[i])
			{
				q.push(Node(i,map[t.to][i]));
			}
		}
		if(cnt == n)
		{
			break;
		}
	}
}


int main()
{
	int n;
	while(scanf("%d",&n) && n != 0)
	{
		for(int i = 0;i < n * (n - 1) / 2;i++)
		{
			int u,v,d;
			scanf("%d%d%d",&u,&v,&d);
			map[u][v] = map[v][u] = d;
		}
		prim(n,1);
		printf("%d\n",ans);
	}
	
	
	return 0;
}

克鲁斯克尔：

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

using namespace std;

struct Node
{
	int from;
	int to;
	int dis;
	Node(){}
	Node(int a,int b,int c)
	{
		from = a;
		to = b;
		dis = c;
	}
	bool operator< (Node t) const
	{
		return dis < t.dis;
	}
};

Node e[20000];
int pre[103]; 
int ans;

int find(int x)
{
	return x == pre[x] ? x : pre[x] = find(pre[x]);
}

void kruskal(int n)
{
	sort(e,e + n * (n - 1) / 2);
	for(int i = 1;i <= n;i++)
	{
		pre[i] = i;
	}
	
	int cnt = 0;
	ans = 0;
	int k = 0;
	while(cnt < n - 1)
	{
		for(int i = k;i < n * (n - 1) / 2;i++)
		{
			int x = find(e[i].from);
			int y = find(e[i].to);
			if(x != y)
			{
				ans += e[i].dis;
				pre[x] = y;
				cnt++;
				k = i + 1;
				break;
			}
		}
	}
}

int main()
{
	int n;
	while(scanf("%d",&n) && n != 0)
	{
		for(int i = 0;i < n * (n - 1) / 2;i++)
		{
			int u,v,d;
			scanf("%d%d%d",&u,&v,&d);
			e[i] = Node(u,v,d);
		}
		kruskal(n);
		printf("%d\n",ans);
	}
	
	
	return 0;
}