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][]=(d[u][]*(d[v][]+d[v][])+d[u][]*d[v][])%mod;
  d[u][]=(d[u][]*(d[v][]+d[v][]))%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+;
 const int mod = ;

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

 vector<int> g[maxn];

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

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

     memset(d,,sizeof(d));

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