Codeforces Round #363 (Div. 2) Problem D - Fix a Tree（并查集 + 构造）

分析：
这题得先分析，每一个连通分量的性质：
- 这个连通分量要么有一个根节点，要么有一个环，且最多只有一个环。

我把这些根节点转入一个容器，把这些形成环的节点（在并查集合并连通分量的时候判断）装入另一个容器。
优先选择已经成为根节点的作为根，否则选择环节点。然后把容器里的其他所有节点连接到根。
心得：
观察性质，简化代码。

#include <iostream>
#include <vector>
const int maxn = int(2e5) + 12;

int par[maxn], G[maxn];

int find(int x) {
    return par[x] == x ? x : par[x] = find(par[x]); 
}

void unite(int a, int b) {
    par[find(a)] = find(b);
}

int main()
{
    //freopen("in.txt", "r", stdin);
    int n, x;
    std::vector<int> my, circle;
    std::cin >> n;
    for (int i = 1; i <= n; i++) par[i] = i; 
    for (int i = 1; i <= n; i++) {
        std::cin >> x;
        G[i] = x;
        if (i == x) my.push_back(i);
        else if (find(x) == find(i)) circle.push_back(i);
        unite(x, i);
    }
    int root;
    int ans = 0;
    if (my.empty()) {
        ans++;
        root = circle[circle.size() - 1];
        G[root] = root;
        circle.pop_back();
    }
    else {
        root = my[my.size() - 1];
        my.pop_back();
    }
    for (auto c : my) {
        ans++;
        G[c] = root;
    }
    for (auto c : circle) {
        ans++;
        G[c] = root;
    }
    std::cout << ans << std::endl;
    for (int i = 1; i <= n; i++) {
        std::cout << G[i] << " ";
    }
    std::cout << std::endl;
    return 0;
}