Codeforces 543 D Road Improvement 樹形dp

題目大意：給定n個點的樹 問：

對於以i爲根時，把樹邊黑白染色，使得任意點走到根的路徑上不超過一條黑邊，輸出染色的方案數(mod 1e9+7)；

題目分析：對於節點u,與其所有聯通的節點v，存在f[u]*=(f[v]+1); 即 當該邊爲白邊時，方案數爲f[v] ，該邊爲黑邊時，方案數爲1；用up[i]來計算當前節點其中一個聯通節點的方案數（其他節點通過他來尋找到當前所要計算的節點）；

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<vector>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long LL;
const LL MOD=1000000007;

vector<int> e[200005];
vector<LL> l[200005],r[200005];
int n,x;
LL f[200005],up[200005];

void dfs1(int u,int fa){
	f[u]=1;
	Fork(p,0,e[u].size()-1){
		int v=e[u][p];
		if (v==fa){
			l[u].push_back(1);
			r[u].push_back(1);
			continue;
		}
		dfs1(v,u);
		f[u]=(f[u]*(f[v]+1))%MOD;
		l[u].push_back(f[v]+1);
		r[u].push_back(f[v]+1);
	}
	For (i,l[u].size()-1){
		l[u][i]=(l[u][i]*l[u][i-1])%MOD;
	}
	RepD(i,r[u].size()-2){
		r[u][i]=(r[u][i]*r[u][i+1])%MOD;
	}	
}

void dfs2(int u,int fa,LL sum){
	up[u]=sum;
	LL tmp;
	Rep (p,e[u].size()){
		int v=e[u][p];
		if (v==fa) continue;
		tmp=sum;
		if (p>0) tmp=(tmp*l[u][p-1])%MOD;
		if (p<e[u].size()-1) tmp=(tmp*r[u][p+1])%MOD;
		dfs2(v,u,tmp+1);
	}
	
	
}

int main(){
	scanf("%d",&n);
	Fork (i,2,n){
		scanf("%d",&x);
		e[x].push_back(i);
		e[i].push_back(x);
	}
	
	dfs1(1,0);
	up[1]=1;
	dfs2(1,0,1LL);
	
	For (i,n){
		printf("%I64d ",f[i]*up[i]%MOD);
	}
}