51 Nod 1500 苹果曼和树（树形dp）

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1500

 

思路：

树形dp。

我们考虑当前结点 i ，对于结点 i ，它可以属于它的子树，也可以不属于，换句话说，我们可以计算出与 i 结点相连的连通块中是否具有黑点的方案数。

于是，d【i】【1】表示包含 i 的连通块中含有黑点，d【i】【0】表示不含有。

接下来考虑状态转移方程：

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

上面的只是我个人认为的。。。可能是错的哈~

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 1e5+5;
const int mod = 1000000007;

int val[maxn];
ll d[maxn][2];

vector<int> g[maxn];

void dfs(int u, int fa)
{
    if(val[u])  d[u][1]=1;
    else d[u][0]=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa)  continue;
        dfs(v,u);
        d[u][1]=(d[u][1]*(d[v][0]+d[v][1])+d[u][0]*d[v][1])%mod;
        d[u][0]=(d[u][0]*(d[v][1]+d[v][0]))%mod;
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    int n;
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x;
        scanf("%d",&x);
        g[x].push_back(i);
        g[i].push_back(x);
    }

    memset(d,0,sizeof(d));

    for(int i=0;i<n;i++)
    {
        scanf("%d",&val[i]);
    }
    dfs(0,-1);
    printf("%lld\n",d[0][1]);
    return 0;
}