[51nod1167]计算递推

题目大意

有这样一个递推式：

F(N) = 1 (0 <= N <= K - 1)
F(N) = F(N - K) + F(N / K) （N % K == 0)
F(N) = F(N - 1) (N % K != 0)

给出2个数K和N，计算F(N)。

2≤K≤10 1≤N≤$k^{50}$

分析

把F(N) = 1 (0 <= N <= K - 1)改成f(0)=1，然后答案一样。

考虑从0转移到N的一个方案，显然它是若干次加一和乘K。

转换一下思路：现在有一个数字集，开始为空。从0转移到N，加1的时候，相当于给集合添加一个1。乘K就相当于数字集里每个数乘K。最终N可以被表示为数字集里所有数的和。很显然每一个数都是K的幂。并且显然一种方案对应了唯一的转移方案。

那么就转换了问题：把N分解成K的幂的和，问有多少种方案。

然后就联想到51nod上的另一道题：整数分解为2的幂
http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1048

那么就可以DP了。设f[i][j]表示把$K^i$分解为最大数是$K^j$的方案数。不过和这道题的区别是，由f[i-1][]得到f[i][]时要合并k块。
DP理解不了请看这里http://blog.csdn.net/worldwide_d/article/details/54091940

要合并K块，就得再套一个DP：设g[j][x]表示把$j*K^i$分成最大数是$K^x$的方案数，然后$g[j][x]=\sum_{x'=0}^{x} g[j-1][x']*f[i-x'-1][x-x']$
最后的$f[i][j]=g[K][j]$
合并答案，就是把N转成K进制，然后剩下的和上面一样。

时间复杂度$O(K\log_{k}^3N)$

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N=50,M=400,mo=1e8;

typedef long long LL;

char c[N];

int tot,m;

LL tt[M];

struct num
{
    int w,a[M];
}f[N+5][N+5],ans,n,p,t,s[N+5],g[N+5][N+5],h[11][N+5];

num operator + (const num &a,const num &b)
{
    t.w=max(a.w,b.w);
    memset(t.a,0,sizeof(t.a));
    for (int i=1;i<=t.w;i++)
    {
        t.a[i]+=a.a[i]+b.a[i];
        if (t.a[i]>=mo)
        {
            t.a[i+1]=1;
            t.a[i]-=mo;
        }
    }
    if (t.a[t.w+1]>0) t.w++;
    return t;
}

num operator * (const num &a,const num &b)
{
    memset(t.a,0,sizeof(t.a));
    memset(tt,0,sizeof(tt));
    for (int i=1;i<=a.w;i++)
    {
        for (int j=1;j<=b.w;j++)
        {
            tt[i+j-1]+=(LL)a.a[i]*b.a[j];
        }
    }
    t.w=a.w+b.w;
    for (int i=1;i<=t.w;i++)
    {
        tt[i+1]+=tt[i]/mo;
        t.a[i]=tt[i]%mo;
    }
    for (;t.w>1 && t.a[t.w]==0;t.w--);
    return t;
}

int main()
{
    scanf("%d%s",&m,c+1);
    int l=strlen(c+1);
    n.w=l/8+1;
    for (int i=1;i<=l;i++)
    {
        int j=(l-i)/8;
        n.a[j+1]=n.a[j+1]*10+c[i]-48;
    }
    f[0][0].w=f[0][0].a[1]=1;
    for (int i=1;i<=N;i++)
    {
        for (int j=0;j<i;j++) g[1][j]=f[i-1][j];
        for (int x=2;x<=m;x++)
            for (int j=0;j<i;j++)
            {
                g[x][j].w=0;
                memset(g[x][j].a,0,sizeof(g[x][j]));
                for (int k=0;k<=j;k++)
                    g[x][j]=g[x][j]+g[x-1][k]*f[i-1-k][j-k];
            }
        for (int j=0;j<i;j++) f[i][j]=g[m][j];
        f[i][i].w=f[i][i].a[1]=1;   
    }
    memset(g,0,sizeof(g));
    g[0][0].w=g[0][0].a[1]=1;
    for (int i=0;i<=N;i++)
    {
        int r=0;
        for (int j=n.w;j;j--)
        {
            int t=n.a[j];
            n.a[j]=((LL)r*mo+t)/m;
            r=((LL)r*mo+t)%m;
        }
        for (int j=0;j<=i;j++) h[0][j]=g[i][j];
        for (int x=1;x<=r;x++)
            for (int j=0;j<=i;j++)
            {
                h[x][j].w=0;
                memset(h[x][j].a,0,sizeof(h[x][j].a));
                for (int k=0;k<=j;k++)
                {
                    h[x][j]=h[x][j]+h[x-1][k]*f[i-k][j-k];
                }
            }
        for (int j=0;j<=i;j++) g[i+1][j]=h[r][j];
    }
    for (int i=0;i<=N;i++) ans=ans+g[N+1][i];
    printf("%d",ans.a[ans.w]);
    for (int i=ans.w-1;i;i--)
    {
        for (int j=1e7;j;j/=10)
        {
            printf("%d",ans.a[i]/j);
            ans.a[i]%=j;
        }
    }
    printf("\n");
    return 0;
}