首页 > 移动平台 > 详细

Codeforces 462D Appleman and Tree 树形dp

时间:2014-08-30 23:11:50      阅读:811      评论:0      收藏:0      [点我收藏+]

题目链接:点击打开链接

题意:

给定n个点的树,

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

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

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

思路:

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


#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;
#define N 300100
#define mod 1000000007
typedef long long ll;
ll dp[N][2]; 
//dp[i][1]表示i点已经归属于某个黑点的方法数
//dp[i][0]表示i点未曾归属某个黑点的方法数
int n, col[N];
vector<int>G[N];
void dfs(int u){
	dp[u][1] = col[u];
	dp[u][0] = col[u]^1;
	for(int i = 0; i < G[u].size(); i++){
		int v = G[u][i];
		dfs(v);
		ll old[2] = {dp[u][0], dp[u][1]};
		dp[u][0] = (old[0]*dp[v][1] +old[0]*dp[v][0])%mod;
		dp[u][1] = (old[1]*dp[v][1] +old[1]*dp[v][0] +old[0]*dp[v][1])%mod;
	}
}
int main() {
    int i, j;
    while(~ scanf("%d",&n)){
		for(i = 0; i < n; i++)G[i].clear();
		for(i = 1; i < n; i++){
			scanf("%d",&j);
			G[j].push_back(i);
		}
		for(i = 0; i < n; i++)scanf("%d",&col[i]);
		dfs(0);
		cout<<dp[0][1]<<endl;
    }
    return 0;
}
/*
5
0 1 1 2
1 0 0 1 1

*/



Codeforces 462D Appleman and Tree 树形dp

原文:http://blog.csdn.net/qq574857122/article/details/38948259

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!