题目大意为:
给你一棵树,其中结点的值不是1就是0,你可以切断一些边,得到一些每棵子树中有且只有1个结点的值为1。
思路:
对于链10001,可知划分数与两个1之间的0的个数有关。
因而我们要把树划成链。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <map>
using namespace std;
const int maxn = 200010;
struct Edge
{
int v, next;
}edge[maxn << 1];
int g[maxn];
int vert[maxn];
int top;
void add(int u, int v)
{
edge[++top].v = v;
edge[top].next = g[u];
g[u] = top;
}
long long ans;
const int mod = 1e9 + 7;
long long int co0[maxn], co1[maxn];
int dfs(int u, int fa)
{
if(vert[u] == 1) co1[u] = 1;
else co0[u] = 1;
for(int i = g[u]; i; i = edge[i].next)
{
int v = edge[i].v;
if(v != fa)
{
dfs(v, u);
co1[u] = (co1[u] * (co0[v] + co1[v]) % mod + co0[u] * co1[v] % mod) % mod;
co0[u] = co0[u] * (co0[v] + co1[v])% mod;
}
}
return 0;
}
int main()
{
int n;
scanf("%d", &n);
for(int i = 1; i < n; i++)
{
int v;
scanf("%d", &v);
add(v, i);
add(i, v);
}
for(int i = 0; i < n; i++)
scanf("%d", &vert[i]);
dfs(0, 0);
printf("%I64d\n", co1[0]);
}