码迷,mamicode.com
首页 > 移动开发 > 详细

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

时间:2017-08-05 06:02:30      阅读:164      评论:0      收藏:0      [点我收藏+]

标签:结果   状态   节点   ios   turn   force   logs   std   max   

链接:

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

题意:

给定n个点的树,

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

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

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

题解:

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

代码:

31 int n;
32 int x[MAXN];
33 VI G[MAXN];
34 ll dp[MAXN][2];
35 
36 void dfs(int u) {
37     if (x[u]) dp[u][1] = 1;
38     else dp[u][0] = 1;
39     rep(i, 0, G[u].size()) {
40         int v = G[u][i];
41         dfs(v);
42         ll old[2] = { dp[u][0], dp[u][1] };
43         dp[u][0] = (old[0] * dp[v][1] + old[0] * dp[v][0]) % MOD;
44         dp[u][1] = (old[1] * dp[v][1] + old[1] * dp[v][0] + old[0] * dp[v][1]) % MOD;
45     }
46 }
47 
48 int main() {
49     ios::sync_with_stdio(false), cin.tie(0);
50     cin >> n;
51     rep(i, 1, n) {
52         int p;
53         cin >> p;
54         G[p].pb(i);
55     }
56     rep(i, 0, n) cin >> x[i];
57     dfs(0);
58     cout << dp[0][1] << endl;
59     return 0;
60 }

 

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

标签:结果   状态   节点   ios   turn   force   logs   std   max   

原文地址:http://www.cnblogs.com/baocong/p/7288245.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!