題目地址：http://codeforces.com/contest/461/problem/B

題目大意：給一棵樹。每一個點為白色或黑色。切斷一些邊，使得每一個連通塊有且僅有一個黑點，問劃分方案數。

算法討論：TreeDP。

f[x][0..1]表示x所在連通塊有0/1個黑點。設y為x的兒子，則DP方程為f[x][1]=f[x][1]*f[y][0]+f[x][1]*f[y][1]+f[x][0]*f[y][1],f[x][0]=f[x][0]*f[y][0]+f[x][0]*f[y][1]。

Code:

#include <cstdio>

#define N 100000
#define mod 1000000007

using namespace std;

long long f[N+10][2];
int n,x,mm,next[N+10],son[N+10],ed[N+10],c[N+10];

inline void add(int x,int y){
	next[++mm]=son[x],son[x]=mm,ed[mm]=y;
}

void dfs(int x){
	f[x][c[x]]=1;
	for (int i=son[x];i;i=next[i]){
		int y=ed[i];
		dfs(y);
		f[x][1]=(f[x][1]*f[y][0]%mod+f[x][1]*f[y][1]%mod+f[x][0]*f[y][1]%mod)%mod;
		f[x][0]=(f[x][0]*f[y][0]%mod+f[x][0]*f[y][1]%mod)%mod;
	}
}

int main(){
	scanf("%d",&n);
	for (int i=2;i<=n;++i) scanf("%d",&x),add(++x,i);
	for (int i=1;i<=n;++i) scanf("%d",&c[i]);
	dfs(1);
	printf("%I64d\n",f[1][1]);
	return 0;
}

By Charlie Pan

Aug 27,2014

Codeforces Round #263 (Div.1) B. Appleman and Tree