L3-035 完美树

思路：

根节点是确定的，所以看到树就想到递归求解。因为每个节点可能有多棵子树，且最终子树都是满足题目要求也就是黑和白最多差一个。所以需要存储一下每棵子树是黑节点多还是白节点多，或者一样多。最终回到父亲节点，判断子树中黑白差，做一个贪心修改最小的（黑白差 / 2 ）个点即可。本题考察我觉得就是模拟 + 递归 + 一丢丢图。下面上代码。代码中也有注释。

代码：

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;
const int N = 100010;

int n;
struct Edge{
    int v, ne;   
}edge[N * 2];
int head[N], idx;
int color[N], val[N];
int res;

void add(int u, int v)
{
    edge[idx].v = v;
    edge[idx].ne = head[u];
    head[u] = idx ++;
}

int dfs(int u, int fa, int &m)
{
    int b = 0, w = 0;
    vector<int> minb, minw;
    for (int i = head[u]; ~i; i = edge[i].ne )
    {
        int v = edge[i].v;
        if (v == fa) continue;
        int minv = 1e5;
        int t = dfs(v, u, minv);
        if (t == 1) // 1代表子树黑的多一个
        {
            b ++;
            minb.push_back(minv);
        }
        else if (t == 0) // 0代表子树白的多一个
        {
            w ++;
            minw.push_back(minv);
        }
    }
    if (color[u] == 1) // 加入根节点
	{
    	b ++; 
    	minb.push_back(val[u]); 
	}
    else 
	{
    	w ++;
    	minw.push_back(val[u]);
	}
    if (b == w) return 2; // 黑白一样多不需要修改，直接返回2
    else 
    {
        sort(minb.begin(), minb.end());
        sort(minw.begin(), minw.end());
        if (b > w) // 判断要修改哪几个节点；
        {
            int t = (b - w) / 2;
            for (int i = 0; i < t; i ++ )
                res += minb[i];
            if ((b - w) % 2) // 差为偶数最终修改完，黑白相等，否则差一个
            {
                m = min(m, minb[t]);
                return 1; // 在这判断最小值
            }
            else return 2;
        }
        else
        {
            int t = (w - b) / 2;
            for (int i = 0; i < t; i ++ )
                res += minw[i];
            if ((w - b) % 2 ) 
            {
                m = min(m, minw[t]);
                return 0;
            }
            else return 2;
        }
    }
}

int main()
{
    cin >> n;
    memset(head, -1, sizeof head);
    for (int i = 1; i <= n; i ++ )
    {
        int k;
        cin >> color[i] >> val[i] >> k;
        for (int j = 0; j < k; j ++ )
        {
            int v; cin >> v;
            add(i, v), add(v, i);
        }
    }
    int m = 1e5;
    dfs(1, -1, m);

    cout << res << endl;
    return 0;
}