题目梗概
有一个n个节点的树,每个节点都有黑色或白色。
问有多少种删边方式,使得删完后的每棵树有且仅有一个黑点。
解题思路
没什么好suo的,直接树形DP。
#include<cstdio>
#define LL long long
using namespace std;
char nc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
if (l==r) return EOF;return *l++;
}
inline int _read(){
int num=0;char ch=nc();
while(ch<'0'||ch>'9') ch=nc();
while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=nc();
return num;
}
const int maxn=100005,tt=1000000007;
int tot,lnk[maxn],nxt[2*maxn],son[2*maxn];
int n,a[maxn];
LL f[maxn][2];
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void DFS(int x,int fa){
f[x][a[x]]=1;
for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa){
DFS(son[j],x);
f[x][1]=(f[x][1]*(f[son[j]][0]+f[son[j]][1])%tt+f[x][0]*f[son[j]][1])%tt;
f[x][0]=f[x][0]*(f[son[j]][1]+f[son[j]][0])%tt;
}
}
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
n=_read();
for (int i=1;i<n;i++) add(_read()+1,i+1);
for (int i=1;i<=n;i++) a[i]=_read();
DFS(1,0);
printf("%lld\n",f[1][1]);
return 0;
}