洛谷 P3155 [CQOI2009]叶子的染色

题目：叶子的染色

思路：
嗯首先每个节点都只有3种状态：白色、黑色、无色。
个人感觉选择根节点不太好处理，那我们先讨论假如根节点确定的情况。
我们先假设一个节点的所有儿子都有颜色，来看一下选择的情况。

如图（日常毁图祭），w代表white，b代表black。

此时可以改变fa以及son的颜色，考虑局部最优的情况：
首先可以知道，儿子的颜色不可改变，但是可以去掉。即一个儿子已经被染成了w，就不能改成b，但是可以变成无色。因为去掉son的颜色还可以通过改变fa来满足c[i]的限制，而改变了颜色一定至少不能满足一个c[i]，且fa无法对叶子节点造成影响。
再来看，此时假如把fa改成w，那么son1的w就没有用了，可以改回无色，此时我们获得了-1+1=0的权。而如果把fa改成b，把son2和son3改回无色，我们获得了-2+1=-1的权。

由于要使得权值最小，所以可以给出一种贪心的策略，即把这个节点改成在儿子中出现次数多的颜色，并把这些儿子变成无色一定是一种最优的情况。
这种策略大概是对于树形dp的变形，把表示状态的一维省略掉。

具体的实现的话dfs一遍就好，先把叶子节点设为c[i]，在回溯的过程中更新答案。
要注意下儿子中白色和黑色的个数相同时，先把父亲设成unknown的颜色，再在接下来的求解过程中让它变成一种最有利的颜色。

接下来是根节点的问题了，结论是任选一个非叶子节点做根均可，但是我不会证明啊QωQ。
嗯可以感性的认识下，就是说假设目前根节点为rt，把根节点迁移到任意一个与rt相邻的节点都不会有权值的改变……好吧说不清楚……
其实这个问题是可以回避的，就是从叶子节点向上推，但是实现起来会困难些。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 100000

int n,m;
int c[maxn+5];
vector<int> g[maxn+5];
int ans=0;

void read(int& x) {
    scanf("%d",&x);
}

int dfs(int x,int fa) {
    if(x<=m) {
        ans++;
        return c[x];
    }
    int w=0,b=0,dk=0;
    for(unsigned int i=0; i<g[x].size(); i++) {
        int y=g[x][i];
        if(y==fa) continue;
        int d=dfs(y,x);
        if(d==0) w++;
        else if(d==-1) dk++;
        else b++;
    }
    if(w<b) {
        ans=ans-(dk+b)+1;
        return 1;
    } else if(w==b) {
        ans=ans-(dk+w)+1;
        return -1;
    } else {
        ans=ans-(dk+w)+1;
        return 0;
    }
}

int main() {
    read(n),read(m);
    for(int i=1; i<=m; i++) {
        read(c[i]);
    }
    for(int i=1; i<n; i++) {
        int x,y;
        read(x),read(y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(m+1,0);
    printf("%d",ans);
    return 0;
}