【AcWing 158周赛】

5842. 拆树

主要思路：

• 如何分割？
  深度遍历，当遍历的子数的sum为总sum的三分之一即是一个分割方法。
  1其中要注意节点范围为$[-100，100]$,所以当$总树sum=0$时，可能可以分成大于3部分。所以最后输出答案时判断条件应该用 $ans.size()>2$
• 这种方法会不会遗漏答案？
  不会，因为从根节点遍历，任意答案方案都能通过上面的方法得到。

输入样例1：

6
2 4
0 5
4 2
2 1
1 1
4 2

输出样例1：

1 4

#include<iostream>
#include<vector>
using namespace std;

const int N = 1e6 + 10;
int d[N], n, h[N], e[N], ne[N], ind = 0;
double sum = 0;
vector<int> ans;

void add(int b, int a) {
    e[++ind] = a;
    ne[ind] = h[b];
    h[b] = ind;
}

void dfs(int r) {
    for (int i = h[r]; i; i = ne[i]) {
        int t = e[i];
        dfs(t);
        d[r] += d[t];
    }
    if (d[r] == sum / 3) {
         ans.push_back(r);
        d[r] = 0;
    }
}
int main() {
     cin >> n;
     int r = 0;
     int x;
     for (int i = 1; i <= n; i++) {
         cin >> x >> d[i];
         if (x) add(x, i);
         else r = i;
         sum += d[i];
     }
    dfs(r);
    if (ans.size() > 2) {
        for (int i =0; i < 2; i++) cout << ans[i] << " ";
    }else puts("-1");
    return 0;
}

5843. 染色

由图可知，所有不连贯的色区边界一定是两色切换的边。
所以，只要遍历整颗树，选两色色块的较小值作为答案。

输入样例1：

11
0 0 0 1 1 0 1 0 0 1 1
1 2
1 3
2 4
2 5
5 6
5 7
3 8
3 9
3 10
9 11

输出样例1：

2

要注意根节点由于没有父节点，所以和根节点连接的块应该额外加上。

输入

15
0 1 0 0 1 1 0 1 1 1 1 1 0 1 0
10 7
10 3
10 8
5 7
13 14
8 13
15 4
15 13
5 2
9 3
11 15
13 6
1 12
9 1

标准答案

3