【问题描述】
雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。
为了灌溉,雷雷需要建立一些水渠,以连接水井和麦田,雷雷也可以利用部分麦田作为“中转站”,利用水渠连接不同的麦田,这样只要一片麦田能被灌溉,则与其连接的麦田也能被灌溉。
现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用(注意不是所有麦田之间都可以建立水渠)。请问灌溉所有麦田最少需要多少费用来修建水渠。
【输入形式】
输入的第一行包含两个正整数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。
思路:构建带权无向图的最小生成树
参考文章:CSP认证201412-4 最优灌溉[C++题解]:最小生成树裸题、Kruskal算法求最小生成树 by 阿正的梦工坊
完整代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int p[N]{};// 用于并查集的数组,p[i]表示i的父节点(或自己)
struct Edge
{
int a,b,c;
};
bool cmp(const Edge &x,const Edge &y)
{
return x.c<y.c;
}
int findroot(int x)//查找x的根节点
{
if (p[x]!=x) p[x]=findroot(p[x]);//路径压缩,减少查找时间
return p[x];
}
int main()
{
int n=0,m=0,re=0;
cin >> n >> m;
for(int i=1;i<=n;i++) p[i]=i;//初始化并查集,每个节点都是自己的根
Edge edges[m];
memset(edges,0,sizeof(edges));
for (int i=0;i<m;i++) cin >> edges[i].a >> edges[i].b >> edges[i].c;
sort (edges,edges+m,cmp);
for (int i=0;i<m;i++)
{
int a=findroot(edges[i].a);
int b=findroot(edges[i].b);
if (a!=b)
{
re+=edges[i].c;
p[a]=b;
}
}
cout << re << endl;
return 0;
}