题目
将最小生成树的条件改一下,让边降序排列进行贪心即可。
教训:WA了2次。
1、题目太简单,没有考虑全面(不连通的情况输出-1)就提交了。
2、判断不连通的条件是第一个点与所有的点连通,但是如果直接用数组p[i]判断父节点的话是不对的,因为并不是所有的点的父节点都是树根。for(int i = 1; i <= N; i++) //WA
if(p[1] != p[i])
最后将判断条件改成:连接的点一共有N-1个则为连通。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int N, M;
const int MAXN = 1010, MAXM = 20010;
int p[MAXN];
struct Edge
{
int u, v, w;
bool operator<(const Edge e){return w > e.w;}
}edge[MAXM];
int find(int x) { return p[x] == x ? x : p[x] = find(p[x]);}
int Kruskal()
{
int ans = 0, tot = 0;
for(int i = 1; i <= N; i++) p[i] = i;
sort(edge, edge+M);
for(int i = 0; i < M; i++)
{
Edge e = edge[i]; e.u = find(e.u); e.v = find(e.v);
if(e.u != e.v)
{
ans += e.w;
p[e.u] = e.v;
tot++;
}
}
//for(int i = 1; i <= N; i++)
// if(p[1] != p[i])
if(tot != N-1)
return -1;
return ans;
}
int main()
{
//freopen("in.txt", "r", stdin);
scanf("%d%d", &N, &M);
for(int i = 0; i < M; i++)
scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].w);
printf("%d\n", Kruskal());
return 0;
}