codeforces 461B 树形dp

做法：

dp[i][0]表示以节点 i 为根的子树没有黑点的方案数，dp[i][1]表示以节点 i 为根的子树中有且仅有一个黑点的方案数。
转移的办法是对于这个树上的每一个节点，如果这个节点是白点，那么就会发现它的子树对dp[i][0]的影响就是不断把（dp[ e[i].to ][0]+dp[ e[i].to ][1]）累乘起来，因为如果 i 的一个子树是个全白的树的话就是连边，不是全白就切断，所以都是可以贡献答案的，但是dp[ i ][1]的话，就要考虑做一个线性的dp，因为要求有且仅有一个黑点，所以选取 i 的子树的时候不能全选全白点的也不能选多个有一个黑点的，所以可以线性地递推一下，pre1表示之前选过有黑点的子树，pre0表示之前没有选过有黑点的子树的方案数。
如果 i 是黑点的话就很简单了，因为如果一个子树是白的，那就可以连，如果是黑的，那就可以断，所以每次乘上（dp[ e[i].to ][0]+dp[ e[i].to ][1]）就可以了。

#include<bits/stdc++.h>
using namespace std;
const int maxn=100005;
const long long mod=1e9+7;
struct edge{
    int to,nxt;
}e[maxn*2];
int ecnt,head[maxn];
int col[maxn];
void ins(int u,int v){
    e[ecnt].to=v;
    e[ecnt].nxt=head[u];
    head[u]=ecnt++;
}
long long dp[maxn][2];
void dfs(int u,int fa){
    if(col[u]){
        long long tmp1=1;
        for(int i=head[u];~i;i=e[i].nxt){
            if(e[i].to==fa) continue;
            dfs(e[i].to,u);
            tmp1=(tmp1*(dp[e[i].to][0]+dp[e[i].to][1]))%mod;
        }
        dp[u][1]=tmp1;
    }else{
        long long tmp1=1,pre0=1,pre1=0;
        for(int i=head[u];~i;i=e[i].nxt){
            if(e[i].to==fa) continue;
            dfs(e[i].to,u);
            tmp1=(tmp1*(dp[e[i].to][0]+dp[e[i].to][1]))%mod;
            if(!pre1) pre1=dp[e[i].to][1];
            else pre1=(pre0*dp[e[i].to][1]%mod+pre1*(dp[e[i].to][0]+dp[e[i].to][1])%mod)%mod;
            pre0=((dp[e[i].to][0]+dp[e[i].to][1])*pre0)%mod;
        }
        dp[u][0]=tmp1;
        for(int i=head[u];~i;i=e[i].nxt){
            if(e[i].to==fa) continue;
            dp[u][1]=pre1;
        }
    }
}
int main(){
    int n,x;
    scanf("%d",&n);
    for(int i=0;i<n;i++) head[i]=-1;
    for(int i=0;i<n-1;i++) {scanf("%d",&x);ins(x,i+1),ins(i+1,x);}
    for(int i=0;i<n;i++) scanf("%d",&col[i]);
    dfs(1,-1);
    printf("%lld\n",dp[1][1]);
    return 0;
}