题目大意
有这样一个递推式:
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≤ k50
分析
把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]表示把
Ki
分解为最大数是
Kj
的方案数。不过和这道题的区别是,由f[i-1][]得到f[i][]时要合并k块。
DP理解不了请看这里http://blog.csdn.net/worldwide_d/article/details/54091940
要合并K块,就得再套一个DP:设g[j][x]表示把
j∗Ki
分成最大数是
Kx
的方案数,然后
g[j][x]=∑xx′=0g[j−1][x′]∗f[i−x′−1][x−x′]
最后的
f[i][j]=g[K][j]
合并答案,就是把N转成K进制,然后剩下的和上面一样。
时间复杂度 O(Klog3kN)
#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;
}