【第12题】[NOIP2015 提高组] 信息传递

题目：[NOIP2015 提高组] 信息传递

题目原文请移步下面的链接

• https://www.luogu.com.cn/problem/P2661
  • 参考题解：https://www.luogu.com.cn/problem/solution/P2661
• 标签：OI、 NOIP、并查集、DFS
• 难度：普及/提高-

题解

思路

• 题解大家可移步看这里，很多童鞋写了各种解法
  • https://www.luogu.com.cn/problem/solution/P2661
• 大体上就是并查集，合并过程中同时需要传递到根节点的距离（其实就是找最小环）
• 当有两个点根节点一致时计算距离，并更新最小环长度

代码

#include <bits/stdc++.h>
using namespace std;
#define endl '\n';

struct edge {
    int r, h;
};

struct UF {
    vector<edge> fa;
    vector<int> m;

    UF(int n) :
            fa(n + 1), m(n + 1) {}

    void initialize (int n) {
        for (int i = 1; i <= n; ++i) {
            cin >> m[i];
            fa[i].r = i;
        }
    }

    int Find(int a) {
        if (a != fa[a].r) {
            int r = fa[a].r;
            fa[a].r = Find(fa[a].r);
            fa[a].h += fa[r].h;
        }
        return fa[a].r;
    }

    void Union(int x, int y, int &solve) {
        int b = Find(y);
        int a = Find(x);
        if (a != b) {
            fa[a].r = b;
            fa[x].h = fa[y].h + 1;
        } else {
            ans = min(solve, fa[x].h + fa[y].h + 1);
        }
    }
};

void best_coder() {
    int n;
    cin >> n;
    UF uf(n);
    uf.initialize(n);
    int solve = 0x7f7f7f7f;
    for (int i = 1; i <= n; ++i) {
       uf.Union(i, uf.m[i], solve);
    }

    cout << solve;
}

int main() {
    // 提升cin、cout效率
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    // 小码匠
    best_coder();

    // 返回
    return 0;
}