您的位置:首页 > 其它

3696: 化合物|树形DP

2016-02-28 16:49 423 查看
用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
,lst
,nxt
;
int mx
,a
[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