前言:
曲神:今天中午做了个噩梦,梦见给DP讲题,讲了半个小时。。。
分析:
终于在曲神,ZYXZYXZYX和舒老师的共同努力下,zz博主终于学会了70%做法
动态规划
我们首先就要设计状态:
f[i][j]:表示到第i位为止,[i-j+1,i]这j位每一位上的颜色都不同(一共有j种颜色),这种情况下的方案数
我们需要从f[i-1]转移到f[i]
有两种情况:
(色差盲福利)
得到的状态转移:
要注意,不能有连续m种不同颜色,所以第二维只能到m-1
这样我们就可以转移了
为了提高效率,我们考虑用矩阵加速:
我们处理出f[1]~f[m-1]
之后的n-m+1项,我们用矩阵加速
//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
using namespace std;
const ll p=1000000007;
ll n,f[1001][201],ans[1001];
int m;
struct node{
ll H[101][101];
node operator *(const node &a) const
{
node ans;
for (int i=1;i<m;i++)
for (int j=1;j<m;j++)
{
ans.H[i][j]=0;
for (int k=1;k<m;k++)
ans.H[i][j]=(ans.H[i][j]%p+H[i][k]*a.H[k][j]%p)%p;
}
return ans;
}
node clear()
{
memset(H,0,sizeof(H));
}
node KSM(ll p)
{
node a=(* this);
node ans=(* this);
p--;
while (p)
{
if (p&1) ans=ans*a;
p>>=1;
a=a*a;
}
return ans;
}
};
node H;
ll KSM(ll a,ll b)
{
ll t=1LL;
a%=p;
while (b)
{
if (b&1)
t=(t%p*a%p)%p;
b>>=1;
a=(a%p*a%p)%p;
}
return t%p;
}
int main()
{
freopen("herzlich.in","r",stdin);
freopen("herzlich.out","w",stdout);
scanf("%lld%d",&n,&m);
if (m>n)
{
printf("%lld",KSM(m,n)%p);
return 0;
}
for (int i=1;i<m;i++)
for (int j=1;j<=i;j++)
H.H[i][j]=1;
for (int i=1;i<m-1;i++) H.H[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)%p;
for (int k=j;k<m;k++)
f[i][j]=(f[i][j]+f[i-1][k])%p;
}
for (int k=1;k<m;k++) f[i][1]=(f[i][1]+f[i-1][k])%p;
}
node an=H.KSM(n-(ll)(m-1));
for (int i=1;i<m;i++)
for (int k=1;k<m;k++)
ans[i]=(ans[i]+f[m-1][k]*an.H[k][i])%p;
ll sum=0;
for (int i=1;i<m;i++) sum=(sum+ans[i])%p;
printf("%lld\n",sum);
return 0;
}