最优灌溉(csp201412-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
问题分析
解题思路
读了好几遍,才发现是kruskal的题。直接套模板即可。需要注意后面数据范围较大,并查集需要使用路径压缩。
参考代码
#include <iostream>
#include <queue>
using namespace std;
class edge
{
public:
int u,v,w;
edge(int from,int to,int weight)
{
u=from;
v=to;
w=weight;
}
bool operator < (const edge& e) const
{
return w>e.w;
}
};
int n,m;
priority_queue<edge> pq;
int p[1010];
void init()
{
while(!pq.empty()) pq.pop();
for(int i=0;i<1010;i++)
{
p[i]=i;
}
}
int find(int a)
{
if(p[a]!=a) p[a]=find(p[a]);
return p[a];
}
int kruskal()
{
int ans=0;
int cnt=0;
while(cnt!=n-1)
{
edge t=pq.top();
pq.pop();
int a=find(t.u);
int b=find(t.v);
if(a==b) continue;
p[a]=b;
ans+=t.w;
cnt++;
}
return ans;
}
int main()
{
init();
scanf("%d %d",&n,&m);
for(int i=1;i<=m;i++)
{
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
edge e(a,b,c);
pq.push(e);
}
printf("%d",kruskal());
return 0;
}
心得体会
非常常规的一道kruskal练习题,算是又复习了一遍最小生成树了。