T2
题解:
方法一:现(da)实(biao)法
我们先写一个dfs找找规律?
1 3 3
2 3 9
3 3 21
4 3 51
5 3 123
6 3 297
7 3 717
…
看上去毫无规律啊
但是我们有充足的想象!
21-9-9=3
51-21-21=9
123-51-51=21
297-123-123=51
717-297-297=123
…
m=3:f[i]=2*f[i-1]+f[i-2]
嗯我们看看4
2 4 16
3 4 64
4 4 232
5 4 856
6 4 3160
7 4 11656
m=4:f[i]=3*f[i-1]+2*f[i-2]+2*f[i-3]
1 5 5
2 5 25
3 5 125
4 5 625
5 5 3005
6 5 14545
7 5 70445
8 5 341185
m=5:f[i]=4*f[i-1]+3*f[i-2]+4*f[i-3]+6*f[i-4]
哇看起来完全没有什么关系,但是rty dalao强行找到了规律
这样只需要加个矩乘优化就好了
方法二:科学dp法
f[i][j]:表示到第i位为止,[i-j+1,i]这j位每一位上的颜色都不同(一共有j种颜色),这种情况下的方案数
我们需要从f[i-1]转移到f[i]
此处引用C_T的图辣
不能有连续m种不同颜色,所以第二维只能到m-1
我们处理出f[1]~f[m-1],之后的n-m+1项,我们用矩阵加速
C_T的图也很好呀,在矩乘的时候最靠左的一行是横着的就是了
于是我们得到了二维递推数组矩乘的方法:按照第一维为一行,第二维递增,最后全部累加
代码:
法一:
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
const int mod=1e9+7;
LL n,f[205],fac[205];int m;
struct mat{int sq[200][200];}mul;
mat calc(mat a,mat b)
{
mat c;memset(c.sq,0,sizeof(c.sq));
for (int i=1;i<=m-1;i++)
for (int j=1;j<=m-1;j++)
for (int k=1;k<=m-1;k++)
c.sq[i][j]=(c.sq[i][j]+((LL)a.sq[i][k]*b.sq[k][j])%mod)%mod;
return c;
}
mat ksm(mat a,LL k)
{
mat ans=a; k--;
for (;k;k>>=1,a=calc(a,a))
if (k&1) ans=calc(ans,a);
return ans;
}
int main()
{
freopen("herzlich.in","r",stdin);
freopen("herzlich.out","w",stdout);
scanf("%lld%d",&n,&m);
f[0]=1; fac[0]=1;
for (int i=1;i<=m;i++)
{
f[i]=f[i-1]*(LL)m%mod;
fac[i]=fac[i-1]*(LL)i%mod;
}
if (n<m) {printf("%lld",f[n]);return 0;}
for (int i=1;i<=m-2;i++) mul.sq[i][i+1]=1;
for (int i=1;i<=m-1;i++) mul.sq[m-1][i]=((LL)i*fac[m-i-1])%mod;
mul=ksm(mul,n-m+1);
LL ans=0;
for (int i=1;i<=m-1;i++) ans=(ans+f[i]*(LL)mul.sq[m-1][i]%mod)%mod;
printf("%lld",ans);
}
法二:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const ll Mod=1e9+7;
ll n,F[1001][201],answer[1001];int m;
struct mat{ll t[120][120];}H;
mat mul(mat a,mat b){
mat C;
for(int i=1;i<m;i++)
for(int j=1;j<m;j++){
C.t[i][j]=0;
for(int k=1;k<m;k++)
C.t[i][j]=(C.t[i][j]+a.t[i][k]*b.t[k][j])%Mod;
}
return C;
}
mat ksm(mat A,ll b){
mat ans=A;b--;
for(;b;b>>=1,A=mul(A,A))
if(b&1) ans=mul(ans,A);
return ans;
}
int main()
{
freopen("herzlich.in","r",stdin);
freopen("herzlich.out","w",stdout);
scanf("%lld%d",&n,&m);
for(int i=1;i<m;i++)
for(int j=1;j<=i;j++)
H.t[i][j]=1;
for(int i=1;i<m-1;i++) H.t[i][i+1]=m-i;
F[1][1]=m;
for(int i=2;i<m;i++){
for(int j=2;j<m;j++){
F[i][j]=F[i-1][j-1]*(m-j+1)%Mod;
for(int k=j;k<m;k++)
F[i][j]=(F[i][j]+F[i-1][k])%Mod;
}
for(int k=1;k<m;k++) F[i][1]=(F[i][1]+F[i-1][k])%Mod;
}
if(n<=m-1){
ll sum=0;
for(int i=1;i<m;i++) sum=(sum+F[n][i])%Mod;
printf("%lld\n",sum);return 0;
}
H=ksm(H,n-m+1);
ll sum=0;
for(int i=1;i<m;i++)
for(int k=1;k<m;k++)
sum=(sum+F[m-1][k]*H.t[k][i])%Mod;
printf("%lld\n",sum);
}