51 nod 1500 苹果曼和树（树形DP）

1500 苹果曼和树

题目来源： CodeForces

基准时间限制：1 秒 空间限制：131072 KB 分值: 80 难度：5级算法题

 收藏

 关注

苹果曼有一棵n个点的树。有一些（至少一个）结点被标记为黑色，有一些结点被标记为白色。

现在考虑一个包含k(0 ≤ k < n)条树边的集合。如果苹果曼删除这些边，那么会将这个树分成(k+1)个部分。每个部分还是一棵树。

现在苹果曼想知道有多少种边的集合，可以使得删除之后每一个部分恰好包含一个黑色结点。答案对1000000007 取余即可。

Input

单组测试数据。第一行有一个整数n (2 ≤ n ≤ 10^5)，表示树中结点的数目。第二行有n-1个整数p[0],p[1],...,p[n-2] (0 ≤ p[i] ≤ i)。表示p[i]和(i+1)之间有一条边。结点从0开始编号。第三行给出每个结点的颜色，包含n个整数x[0],x[1],...,x[n-1] (x[i]是0或者1)。如果x[i]是1，那么第i个点就是黑色的，否则是白色的。

Output

输出答案占一行。

Input示例

30 00 1 1

Output示例

2

这题的操作太tmd骚了 学不来学不来

这题的实质就是想让你求删除边的方法 使每一个联通块仅包含一个黑点

如果一个节点是白色那么他一定要连接到黑色联通块 

 dp[u][0]=(dp[u][0]*(dp[v][0]+dp[v][1]))%mod;（加上子节点的黑色联通块是因为 子节点有多少种情况形成黑色联通块 删除这条边就变成白色联通块 子节点分割不同的方法数）

dp[u][1]=(dp[u][1]*(dp[v][0]+dp[v][1])+dp[u][0]*dp[v][1])%mod;

#include <iostream>#include <stdio.h>#include <string.h>#include <stack>#include <queue>#include <map>#include <set>#include <vector>#include <math.h>#include <bitset>#include <algorithm>#include <climits>using namespace std;const int N = 200000+100;typedef long long LL;const LL mod = 1000000007;typedef long long LL;vector<int>p[N];LL dp[N][2], a[N];void dfs(int u,int fa){    if(a[u]) dp[u][1]=1;    else dp[u][0]=1;    for(int i=0;i<p[u].size();i++)    {        int v=p[u][i];        if(v==fa) continue;        dfs(v,u);        dp[u][1]=(dp[u][1]*(dp[v][0]+dp[v][1])+dp[u][0]*dp[v][1])%mod;        dp[u][0]=(dp[u][0]*(dp[v][0]+dp[v][1]))%mod;    }    return ;}int main(){    int n;    scanf("%d", &n);    for(int i=1;i<n;i++)    {        int x;        scanf("%d", &x);        p[x].push_back(i),p[i].push_back(x);    }    for(int i=0;i<n;i++) scanf("%lld", &a[i]);    memset(dp,0,sizeof(dp));    dfs(0,-1);    cout<<dp[0][1]<<endl;    return 0;}