[BZOJ1089][SCOI2003]严格n元树（dp+数学相关+高精度）

题目描述

传送门

题解

刚开始一直在考虑如何在一棵树上新挂一些节点，实际上我们可以反过来考虑，将一些已经构造好的子树连到一个根上，也就是从下往上堆
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]);
}