[CodeForces 1375G] Tree Modification（转化） | 错题本

题目

[CodeForces 1375G] Tree Modification

分析

如果对树黑白染色，题目的操作方式与将一个点（即 $a$ 点）变色等价，目标状态就是只有一个白色节点或一个黑色节点。因此记原树中白色节点个数为 $c_1$，黑色节点个数为 $c_2$，输出 min ⁡ { c 1 , c 2 } − 1 \min\{c_1, c_2\} - 1 min{c1​,c2​}−1 即可。

代码

#include <bits/stdc++.h>

const int MAXN = 500000;
const int MAXM = 1000000;

int N, Cnt[2];
std::vector<int> G[MAXN + 5];

void Dfs(int u, int f, int c) {
    Cnt[c]++;
    for (int i = 0; i < (int)G[u].size(); i++) {
        int v = G[u][i];
        if (v != f)
            Dfs(v, u, c ^ 1);
    }
}

int main() {
    scanf("%d", &N);
    for (int i = 1; i < N; i++) {
        int u, v; scanf("%d%d", &u, &v);
        G[u].push_back(v), G[v].push_back(u);
    }
    Dfs(1, 0, 0);
    printf("%d", std::min(Cnt[0], Cnt[1]) - 1);
    return 0;
}