Codeforces 461B Appleman and Tree

树形DP

dp[i][0]表示不包括i的联通快没有黑点的方案数

dp[i][1]表示不包括i的联通块有黑点的方案数

这样对于树根i，每一个插进来的子叶j，都可以判断dp[j][0]和dp[j][1]对于dp[i][0]和dp[i][1]的贡献

列如两个联通块都有黑点，那么这条相连的边就要割掉

代码：

#include<bits/stdc++.h>
#define pb push_back
#define fr first
#define sc second
using namespace std;

typedef long long ll;
typedef pair<ll, ll> PLL;

const int N = 1e5+5;
const ll MOD = 1e9+7;

vector<int> G[N];
ll dp[N][2];
int x[N];

void dfs(int u, int fa) {
    if(x[u]) dp[u][1] = 1, dp[u][0] = 0;
    else dp[u][1] = 0, dp[u][0] = 1;
    for(auto v:G[u]) if(v != fa) {
        dfs(v, u);
        ll t1 = dp[u][1], t0 = dp[u][0];
        dp[u][1] = ((t0*dp[v][1])%MOD+(t1*dp[v][1])%MOD+(t1*dp[v][0])%MOD)%MOD;
        dp[u][0] = ((t0*dp[v][0])%MOD+(t0*dp[v][1])%MOD)%MOD;
    }
}

int main() {
    int n, u;
    scanf("%d", &n);
    for(int v = 1; v < n; v++) {
        scanf("%d", &u);
        G[u].pb(v);
    }
    for(int i = 0; i < n; i++) scanf("%d", &x[i]);
    dfs(0, -1);
    printf("%lld\n", dp[0][1]);
    return 0;
}