Codeforces 698B Fix a Tree

发表于 分类于 Disqus：

链接

传送门

题意

给出一个\(n\)个结点的父结点，改变最少的结点的父结点使得图变为一棵有向树。

思路

dfs搜出所有的环，指定新的根，把环解开后连到根上就好啦。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int maxn = 200005;

int a[maxn];
int x[maxn], vis[maxn];

void dfs(int u, int id) {
  vis[u] = id;
  if (vis[a[u]]) {
    x[id] = vis[a[u]] == id ? u : -1;
  } else {
    dfs(a[u], id);
  }
}

int main() {
  int n, u = 0;
  scanf("%d", &n);
  for (int i = 1; i <= n; ++i) {
    scanf("%d", &a[i]);
    if (a[i] == i) {
      u = i;
    }
  }
  int num = 0;
  for (int i = 1; i <= n; ++i) {
    if (!vis[i]) {
      dfs(i, ++num);
    }
  }
  if (u == 0) {
    for (int i = 1; i <= num; ++i) {
      if (x[i] != -1) {
        u = x[i];
        break;
      }
    }
  }
  int ans = 0;
  for (int i = 1; i <= num; ++i) {
    if (x[i] != -1 && a[x[i]] != u) {
      a[x[i]] = u;
      ++ans;
    }
  }
  printf("%d\n", ans);
  for (int i = 1; i <= n; ++i) {
    printf("%d ", a[i]);
  }
  puts("");
  return 0;
}
相关文章