poj 2377 Bad Cowtractors(最小生成树)

还是简单的的MST（最小生成树）

数据大意:

N，M 表示图有N个顶点，M条边。

下面有M组数据，a，b，c

依次表示顶点a到b的权值是c

这个题要求的并不是MST，而是要生成树的边的权值总和最大，其实就是MST的变形

不同是多了一个条件，若图中不存在MST，则输出-1，这里没有看到，wrong了一次。

因为题目直接给出的是边的信息，所以这里就用了Kruscal算法

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
const int vMax = 1010;

struct Edge
{
    int from,to,w;
    Edge(int a,int b,int c)
    {
        from = a;
        to = b;
        w = c;
    }
    friend bool operator < (const Edge& a,const Edge& b)
    {
        return a.w > b.w;
    }
};

vector<Edge>v;
int N,M;
int root[vMax];

void init_set(int n)
{
    for(int i = 1 ; i <= n ; i++)
    {
        root[i] = i;
    }
}

int find_set(int a)
{
    if(root[a] != a)
    {
        root[a] = find_set(root[a]);
    }
    return root[a];
}

void union_set(int a,int b)
{
    int x = find_set(a);
    int y = find_set(b);
    if(x != y)
    {
        root[x] = y;
    }
}

void MST()
{
    int ans = 0;
    for(int i = 0 ; i < v.size() ; i++)
    {
        int a = v[i].from;
        int b = v[i].to;
        int c = v[i].w;
        if(find_set(a) != find_set(b))//可以合并，合并到同一子树中
        {
            ans += c;//总权值不断++
            union_set(a,b);
        }
    }
    int k = find_set(1);//判断图是否连通
    bool flag = false;
    for(int i = 2 ; i <= N ; i++)
    {
        if(find_set(i) != k)
        {
            flag = true;
        }
    }
    if(flag)
    {
        ans = -1;
    }
    printf("%d\n",ans);
}
int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif // LOCAL
    while(scanf("%d %d",&N,&M) != EOF)
    {
        init_set(N);
        int a,b,c;
        v.clear();
        for(int i = 0 ; i < M ; i++)
        {
            scanf("%d %d %d",&a,&b,&c);
            v.push_back(Edge(a,b,c));
        }
        sort(v.begin(),v.end());//边的权值按照降序排序，这样求出来的生成树的权值是最大的
        MST();
    }
    return 0;
}