比赛时就知道是树形dp但是和一般的熟悉的树形背包有区别。
dp[i][0]表示以i为根节点的树中没有黑色节点的数量。
dp[i][1]表示以i为根节点的树中有1个黑色节点的数量。
代码语言：javascript复制#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<math.h>
#include<queue>
using namespace std;
const int mod=1e9 7;
int n,color[100005];
vector <int> map[100005];
long long dp[100005][2];
void dfs(int now)
{
dp[now][0]=1;
dp[now][1]=0;
for(int i=0;i<map[now].size();i )
{
int to=map[now][i];
dfs(to);
dp[now][1]=(dp[now][1]*dp[to][0] dp[now][0]*dp[to][1])%mod;
dp[now][0]=dp[now][0]*dp[to][0]%mod;
}
if(color[now]==1)
dp[now][1]=dp[now][0];
else
dp[now][0]=(dp[now][0] dp[now][1])%mod;
}
int main()
{
int n;
cin>>n;
for(int i=1;i<n;i )
{
int a;
scanf("%d",&a);
map[a].push_back(i);
}
for(int i=0;i<n;i )
scanf("%d",&color[i]);
dfs(0);
long long ans=dp[0][1];
cout<<ans<<endl;
return 0;
}
