题目:叶子的染色
思路:
嗯首先每个节点都只有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;
}