题目描述
题解
刚开始一直在考虑如何在一棵树上新挂一些节点,实际上我们可以反过来考虑,将一些已经构造好的子树连到一个根上,也就是从下往上堆
f(i)表示深度为i的严格n元树的数量,s(i)表示f(i)的前缀和;那么f(i)=s(i-1)^n-s(i-2)^n,也就是说n个位置每个位置都可以接一个子树,但是要保证所有的子树里至少有一个深度为i-1
代码
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
int n,d;
struct data{int a[1005];}zero,one,f[50],s[50],mi[50];
void print(data a)
{
for (int i=a.a[0];i>=1;--i)
{
if (i==a.a[0]||a.a[i]>=1000) printf("%d",a.a[i]);
else if (a.a[i]>=100) printf("0%d",a.a[i]);
else if (a.a[i]>=10) printf("00%d",a.a[i]);
else printf("000%d",a.a[i]);
}
puts("");
}
data jia(data a,data b)
{
data ans=zero;int len=max(a.a[0],b.a[0]);
for (int i=1;i<=len;++i) ans.a[i]=a.a[i]+b.a[i];
for (int i=1;i<=len;++i)
{
ans.a[i+1]+=ans.a[i]/10000;
ans.a[i]%=10000;
}
if (ans.a[len+1]) ++len;
ans.a[0]=len;
return ans;
}
data jian(data a,data b)
{
data ans=zero;int len=a.a[0];
for (int i=1;i<=len;++i)
{
if (a.a[i]<b.a[i])
{
--a.a[i+1];
a.a[i]+=10000;
}
ans.a[i]=a.a[i]-b.a[i];
}
while (len>1&&!ans.a[len]) --len;
ans.a[0]=len;
return ans;
}
data cheng(data a,data b)
{
data ans=zero;int len=a.a[0]+b.a[0]-1;
for (int i=1;i<=a.a[0];++i)
for (int j=1;j<=b.a[0];++j)
ans.a[i+j-1]+=a.a[i]*b.a[j];
for (int i=1;i<=len;++i)
{
ans.a[i+1]+=ans.a[i]/10000;
ans.a[i]%=10000;
}
while (ans.a[len+1])
{
++len;
ans.a[len+1]+=ans.a[len]/10000;
ans.a[len]%=10000;
}
ans.a[0]=len;
return ans;
}
data poww(data a)
{
data ans=one;
for (int i=1;i<=n;++i)
ans=cheng(ans,a);
return ans;
}
int main()
{
scanf("%d%d",&n,&d);
zero.a[0]=1;
one.a[0]=one.a[1]=1;
f[0]=f[1]=one;s[0]=one,s[1]=jia(one,one);
mi[0]=poww(s[0]),mi[1]=poww(s[1]);
for (int i=2;i<=d;++i)
{
f[i]=jia(f[i],mi[i-1]);
f[i]=jian(f[i],mi[i-2]);
s[i]=jia(s[i-1],f[i]);
mi[i]=poww(s[i]);
}
print(f[d]);
}