Barn Painting

$$BarnPainting$$

题目链接：$jzoj6657$ / $luogu4084$

题目

给定一颗$N$个节点组成的树，$3$种颜色，其中$K$个节点已染色，要求任意两相邻节点颜色不同，求合法染色方案数。

输入

第$1$行：$N$和$K$
第$2～N$行：$x$和$y$代表点$x$和点$y$
第$N+1～N+K$行：$b$和$c$代表点$b$已经被涂上了颜色$c$

输出

合法的染色方案数 $\%10^9+7$

样例输入

4 1
1 2
1 3
1 4
4 3

样例输出

8

数据范围

$1\le N\le 10^5$，$0\le K\le N$
$1\le x,y\le N$，$x\ne y$
$1\le b\le N$，$1\le c\le 3$

思路

这道题是一道树形$dp$。
我们设$f[i][j]$为在点$i$涂颜色$j$的方案数，那么首先初始化，我们把所有点涂所有颜色的方案数都是$1$。
接着，我们来处理数据中直接涂了颜色的点。这个也很好做，就是把这个点涂了的颜色的方案数标为$1$，剩下的两个颜色方案数标为$0$。
接着我们就要开始$dp$，就按照与一个点相邻的点不能跟它是同一个颜色的原则，我们就可以写出$dp$式。

注意：

1. 要开$longlong$
2. 要记得取模

代码

#include<cstdio>
#include<cstring>
#define mo 1000000007
#define ll long long

using namespace std;

struct node{
	int to, next;
}e[200001];
int n, k, x, y, kk, le[200001];
ll f[200001][4];

void add(int x, int y) {//连边
	e[++kk] = (node){y, le[x]}; le[x] = kk;
}

void dfs(int now, int fa) {//树形dp
	for (int i = le[now]; i; i = e[i].next)
		if (e[i].to != fa) {
			dfs(e[i].to, now);
			f[now][1] = f[now][1] * ((f[e[i].to][2] + f[e[i].to][3]) % mo) % mo;//三个dp式，分别表示涂这三个颜色的方案数
			f[now][2] = f[now][2] * ((f[e[i].to][1] + f[e[i].to][3]) % mo) % mo;
			f[now][3] = f[now][3] * ((f[e[i].to][1] + f[e[i].to][2]) % mo) % mo;
		}
}

int read() {//快读
	int re = 0, zf = 1;
	char c = getchar();
	while (c < '0' || c > '9') {
		if (c = '-') zf = -zf;
		c = getchar();
	}
	while (c >= '0' && c <= '9') {
		re = re * 10 + c - 48;
		c = getchar();
	}
	return re * zf;
}

int main() {
//	freopen("barnpainting.in", "r", stdin);
//	freopen("barnpainting.out", "w", stdout);
	
	n = read();//读入
	k = read();//读入
	
	for (int i = 1; i <= n; i++)
		f[i][1] = f[i][2] = f[i][3] = 1;//初始化
	
	for (int i = 1; i < n; i++) {
		x = read();//读入
		y = read();//读入
		add(x, y);//连边
		add(y, x);//两个方向又要连边
	}
	for (int i = 1; i <= k; i++) {
		x = read();//读入
		y = read();//读入
		for (int i = 1; i <= 3; i++)
			f[x][i] = 0;//如果一个点已经涂色，那么这个点就不可能涂其他颜色
		f[x][y] = 1;//就只能涂这个颜色
	}
	
	dfs(1, 0);//树形dp
	
	printf("%lld", (f[1][1] + f[1][2] + f[1][3]) % mo);//输出答案
	
//	fclose(stdin);
//	fclose(stdout);
	
	return 0;
}