Description
苹果曼有一棵n个点的树。有一些(至少一个)结点被标记为黑色,有一些结点被标记为白色。
现在考虑一个包含k(0 ≤ k < n)条树边的集合。如果苹果曼删除这些边,那么会将这个树分成(k+1)个部分。每个部分还是一棵树。
现在苹果曼想知道有多少种边的集合,可以使得删除之后每一个部分恰好包含一个黑色结点。答案对1000000007 取余即可。
题解
定义
f[i][0/1]
表示以
i
为根的子树,
考虑如何合并两个子树的答案,可得, f[i][0]∗=f[son][0]+f[son][1];f[i][1]=f[i][0]∗f[son][1]+f[i][1]∗(f[son][0]+f[son][1]) ;
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100006
#define tt 1000000007
#define LL long long
using namespace std;
inline char nc(){
static char buf[100000],*i=buf,*j=buf;
return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;
}
inline int _read(){
char ch=nc();int sum=0;
while(!(ch>='0'&&ch<='9'))ch=nc();
while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
return sum;
}
int n,tot,a[maxn],lnk[maxn],nxt[maxn*2],son[maxn*2];
LL f[maxn][2];
bool vis[maxn];
void add(int x,int y){
nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;
}
void dfs(int x){
vis[x]=0;f[x][a[x]]=1;
for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){
dfs(son[j]);
(f[x][1]=f[x][0]*f[son[j]][1]%tt+f[x][1]*(f[son[j]][0]+f[son[j]][1])%tt)%=tt;
(f[x][0]*=(f[son[j]][1]+f[son[j]][0])%tt)%=tt;
}
}
int main(){
freopen("tree.in","r",stdin);
freopen("tree.out","w",stdout);
n=_read();
for(int i=1,x;i<n;i++)x=_read(),add(i,x),add(x,i);
for(int i=0;i<n;i++)a[i]=_read();
memset(vis,1,sizeof(vis));
dfs(1);
printf("%lld\n",f[1][1]);
return 0;
}