[樹形DP]51 Nod 1500——蘋果曼和樹

題目梗概

有一個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;
}