Codeforces Round #263 (Div. 2) D. Appleman and Tree 树形dp

链接：

http://codeforces.com/contest/462/problem/D

题意：

给定n个点的树，

0为根，下面n-1行表示每个点的父节点

最后一行n个数 表示每个点的颜色，0为白色，1为黑色。

把树分成若干个联通块使得每个联通块有且仅有一个黑点，问有多少种分法（结果mod1e9+7）

题解：

树形dp，每个点有2个状态，已经归属于某个黑点和未归属于某个黑点。

代码：

int n;
int x[MAXN];
VI G[MAXN];
ll dp[MAXN][2];

void dfs(int u) {
    if (x[u]) dp[u][1] = 1;
    else dp[u][0] = 1;
    rep(i, 0, G[u].size()) {
        int v = G[u][i];
        dfs(v);
        ll old[2] = { dp[u][0], dp[u][1] };
        dp[u][0] = (old[0] * dp[v][1] + old[0] * dp[v][0]) % MOD;
        dp[u][1] = (old[1] * dp[v][1] + old[1] * dp[v][0] + old[0] * dp[v][1]) % MOD;
    }
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> n;
    rep(i, 1, n) {
        int p;
        cin >> p;
        G[p].pb(i);
    }
    rep(i, 0, n) cin >> x[i];
    dfs(0);
    cout << dp[0][1] << endl;
    return 0;
}