3696: 化合物|树形DP

a[x][i] 表示以节点 x 为根深度为i的结点数
然后暴力转移……
有人说这是母函数QAQ..smg

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<iostream>
#include<algorithm>
#define N 100052
using namespace std;
int head[N],lst[N],nxt[N];
int mx[N],a[N][505],ans[555];
int n,tot;
void insert(int x,int y)
{
    lst[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
void dfs(int x)
{
    a[x][0]=1;
    for(int i=head[x];i;i=nxt[i])
    {
        dfs(lst[i]);
        for(int j=0;j<=mx[x];j++)
            for(int k=0;k<=mx[lst[i]];k++)
                ans[j^(k+1)]+=a[x][j]*a[lst[i]][k];
        mx[x]=max(mx[x],mx[lst[i]]+1);
        for(int j=0;j<=mx[lst[i]];j++)
            a[x][j+1]+=a[lst[i]][j];
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    {
        int x;scanf("%d",&x);
        insert(x,i);
    }
    dfs(1);
    int MX=512;
    while(!ans[MX-1])MX--;
    for(int i=0;i<MX;i++)printf("%d\n",ans[i]);
    return 0;
}

你可能感兴趣的:(树形DP)