60 lines
1.6 KiB
C++
60 lines
1.6 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
typedef long long LL;
|
||
const LL mod = 1e9 + 7;
|
||
const int N = 1e5 + 10, M = N << 1;
|
||
|
||
int n, m;
|
||
int color[N]; // 值域1~3
|
||
|
||
// 链式前向星
|
||
int e[M], h[N], idx, ne[M];
|
||
void add(int a, int b) {
|
||
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
|
||
}
|
||
|
||
LL f[N][4]; // 设f[u][j]表示将u染色为j时，u这棵子树的方案数
|
||
int st[N]; // 是不是访问过
|
||
|
||
void dfs(int u) {
|
||
st[u] = 1; // 标识已访问
|
||
// 初始化
|
||
if (color[u])
|
||
// 当某个节点被指定上色后，那么该节点另外两种颜色的方案数为0
|
||
// 例如:当点u被指定上色2时：f[u][1]=0,f[u][3]=0 (因为无法上色1和3）
|
||
f[u][color[u]] = 1;
|
||
else
|
||
f[u][1] = f[u][2] = f[u][3] = 1; // 三种颜色都可以染
|
||
|
||
for (int i = h[u]; ~i; i = ne[i]) {
|
||
int v = e[i];
|
||
if (st[v]) continue;
|
||
dfs(v);
|
||
// 对于每个节点，因为不能于子节点上色相同
|
||
f[u][1] = (f[u][1] * (f[v][2] + f[v][3]) % mod) % mod;
|
||
f[u][2] = (f[u][2] * (f[v][1] + f[v][3]) % mod) % mod;
|
||
f[u][3] = (f[u][3] * (f[v][1] + f[v][2]) % mod) % mod;
|
||
}
|
||
}
|
||
|
||
int main() {
|
||
memset(h, -1, sizeof h);
|
||
cin >> n >> m;
|
||
|
||
for (int i = 1; i < n; i++) {
|
||
int a, b;
|
||
cin >> a >> b;
|
||
add(a, b), add(b, a);
|
||
}
|
||
|
||
while (m--) {
|
||
int u, c;
|
||
cin >> u >> c;
|
||
color[u] = c; // u节点被染过色,颜色为c
|
||
}
|
||
|
||
dfs(1);
|
||
|
||
printf("%lld\n", (f[1][1] + f[1][2] + f[1][3]) % mod);
|
||
return 0;
|
||
} |