HDU-3367-Pseudoforest

ACM模版

描述

题解

求最大伪森林，伪森林是最终每个连通分量最多有一个环，可以按照最大生成树的方法求，只需要把克鲁斯卡尔稍微改一下即可，另外加一个flag[]标记是否有环。

真是有句话不假，多个香炉多个鬼，原本把m全局就好了，我的模版偏偏引入了一个tol变量，耗时多不说，还总是忘记给它初始化……下一次真得涨涨记性了，一坑就是半个小时。

代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;

/*
 *  Kruskal算法求MST
 *  对边操作，并排序
 */

const int MAXN = 10010;     //  最大点数
const int MAXM = 100010;    //  最大边数

int F[MAXN];    //  并查集使用
int flag[MAXN]; //  标记是否有环

struct Edge
{
    int u;      //  起点
    int v;      //  终点
    int w;      //  权值
} edge[MAXM];   //  存储边的信息

int tol;        //  边数，加边前赋值为0!!!!!!!

void addEdge(int u, int v, int w)
{
    edge[tol].u = u;
    edge[tol].v = v;
    edge[tol++].w = w;
    return ;
}

bool cmp(Edge a, Edge b)
{
    //  排序函数，将边按照权值从大到小排序
    return a.w > b.w;
}

int find(int x)
{
    if (F[x] == x)
    {
        return x;
    }
    else
    {
        return F[x] = find(F[x]);
    }
}

int Kruskal(int n)  //  传入点数，返回最小生成树的权值，如果不连通则返回-1
{
    for (int i = 0; i <= n; i++)
    {
        F[i] = i;
    }
    sort(edge, edge + tol, cmp);

    int ans = 0;
    for (int i = 0; i < tol; i++)
    {
        int u = edge[i].u;
        int v = edge[i].v;
        int w = edge[i].w;
        int fu = find(u);
        int fv = find(v);
        //  两个都有环则不合并
        if (flag[fu] && flag[fv])
        {
            continue;
        }
        //  合并
        ans += w;
        F[fu] = fv;

        if (flag[fv] || flag[fu] || fv == fu)
        {
            flag[fv] = 1;
        }

    }

    return ans;
}

int main(int argc, const char * argv[])
{
    int n, m;
    int u, v, w;

    while (cin >> n >> m, n != 0 || m != 0)
    {
        tol = 0;
        memset(flag, 0, sizeof(flag));

        for (int i = 0; i < m; i++)
        {
            scanf("%d%d%d", &u, &v, &w);
            addEdge(u, v, w);
        }

        int res = Kruskal(n);
        cout << res << '\n';
    }

    return 0;
}