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

最新推荐文章于 2021-08-08 21:07:49 发布

mrcoderrev

最新推荐文章于 2021-08-08 21:07:49 发布

阅读量332

点赞数

分类专栏：树形DP

本文链接：https://blog.csdn.net/yjf3151731373/article/details/77968913

版权

树形DP 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

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示例

3
0 0
0 1 1

Output示例

这题的操作太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;
}