CCF 201412-4 最优灌溉

题目

问题描述

　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。

输出格式

　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤ci≤10,000。

---

问题分析

一道典型的最小生成树问题，这里使用堆优化的Kruskal算法来求解，堆优化的Prim或Kruskal可以应付绝大多数的MST问题了~。因为使用的是Kruskal算法，所以涉及到了不相交集（并查集），很简单，为提高效率，并查集采用了路径压缩和按树大小合并，去掉应该也无伤大雅。然后就是priority_queue的使用方法了（STL还是好用，自己实现的堆只拿了20分= =）。

代码

#include<cstdio>
#include<queue>
using namespace std;

const int MAX_V = 1000;

struct edge{
	int front,to;
	int weight;
	bool operator < (const edge &a) const{
		return weight > a.weight;
	}
};

int p[MAX_V+1];

void InitialDisjSet(){
	for(int i=0;i<=MAX_V;i++)
		p[i] = -1;
}

int Find(int x){
	if(p[x]>=0)
		return p[x] = Find(p[x]);
	else
		return x;
}

void Union(int x,int y){
	int xroot = Find(x), yroot = Find(y);
	if(xroot!=yroot){
		if(p[xroot]<=p[yroot]){
			p[xroot] += p[yroot];
			p[yroot] = xroot;	
		}
		else{
			p[yroot] += p[xroot];
			p[xroot] = yroot;
		}
	}
}

int kruskal(priority_queue<edge> &q,int n){
	edge e;
	int ret = 0;
	InitialDisjSet();
	for(int i=1;i<n;i++){
		e = q.top();
		q.pop();
		if(Find(e.front)!=Find(e.to)){
			ret += e.weight;
			Union(e.front,e.to);
		}
		else
			i--;
	}
	return ret;
}

int main(){
	int n,m;
	priority_queue<edge> q;
	scanf("%d %d",&n,&m);
	for(int i=0;i<m;i++){
		edge e;
		scanf("%d %d %d",&e.front,&e.to,&e.weight);
		q.push(e);
	}
	printf("%d\n",kruskal(q,n));
	
	return 0;
}