Codeforces 461B Appleman and Tree

http://codeforces.com/problemset/problem/461/B

思路:dp,dp[i][0]代表这个联通块没有黑点的方案数，dp[i][1]代表有一个黑点的方案数

转移:
首先如果儿子节点是个有黑点的，父亲节点也有黑点，那么只能分裂开，不能合并在一起，只对dp[u][1]有贡献

      如果儿子节点是个有黑点的，父亲节点没有黑点，那么可以分裂也可以合并，对dp[u][1]和dp[u][0]都有贡献

      如果儿子节点是个没有黑点的，父亲节点有黑点，那么必须和父亲合并，对dp[u][1]有贡献

      如果儿子节点是个没有黑点的，父亲节点也没黑点，那么必须和父亲合并，对dp[u][0]有贡献

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<iostream>
#define ll long long
const ll Mod=1000000007;
ll f[200005][2];
int v[200005],n,tot,go[400005],first[200005],next[400005];
int read(){
    int t=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();}
    return t*f;
}
void insert(int x,int y){
    tot++;
    go[tot]=y;
    next[tot]=first[x];
    first[x]=tot;
}
void add(int x,int y){
    insert(x,y);insert(y,x);
}
void dfs(int x,int fa){
    if (v[x]==1) f[x][1]=1,f[x][0]=0;else f[x][1]=0,f[x][0]=1;
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        dfs(pur,x);
        ll t0=f[x][0],t1=f[x][1];
        f[x][1]=((t0*f[pur][1])%Mod+(t1*f[pur][0])%Mod)+(t1*f[pur][1])%Mod;
        f[x][0]=((t0*f[pur][0])%Mod+(t0*f[pur][1])%Mod);
    }
}
int main(){
    n=read();
    for (int i=2;i<=n;i++){
        int x=read()+1;
        add(x,i);
    }
    for (int i=1;i<=n;i++) v[i]=read();
    dfs(1,0);
    printf("%I64d\n",(f[1][1])%Mod);
    return 0;
}