计数
题目
分析
既然,位数很大,数位dp。
设\(f_{i,j}\)表示,做到第i位,这一位选什么(0表示这一位选0,1表示这一位选1~k-1)。
转移也就显然了
\[f_{i,0}=f_{i-1,1}
\]
\[f_{i,1}=(f_{i-1,1}+f_{i-1,0})*(k-1)
\]
答案就是\(f_{n,1}+f_{n,0}\)
为了方便处理不能有前导0,从高位做起。
还要打高精度。
#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
const long long maxlongint=2147483647;
const long long mo=100000000;
const long long N=2000;
using namespace std;
long long f[N][2][N],t[1000],n,m;
int add(int x,int pos,int y,int pos1)
{
f[x][pos][0]=max(f[x][pos][0],f[y][pos1][0]);
for(int i=1;i<=f[x][pos][0];i++)
{
f[x][pos][i]+=f[y][pos1][i];
f[x][pos][i+1]+=f[x][pos][i]/mo;
f[x][pos][i]%=mo;
}
if(f[x][pos][f[x][pos][0]+1])
f[x][pos][0]++;
}
int main()
{
scanf("%lld%lld",&n,&m);
f[1][1][0]=1;
f[1][1][1]=m-1;
for(int i=2;i<=n;i++)
{
add(i,0,i-1,1);
for(int j=1;j<=m-1;j++) add(i,1,i-1,1);
for(int j=1;j<=m-1;j++) add(i,1,i-1,0);
}
add(n,1,n,0);
printf("%lld",f[n][1][f[n][1][0]]);
for(int i=f[n][1][0]-1;i>=1;i--)
{
printf("%08lld",f[n][1][i]);
}
}