bzoj5190 [Usaco2018 Jan]Stamp Painting（dp+计数）

一共有n个格子，m种颜色，你每次可以
把连续K个格子涂成一种颜色。问你最后有多少种
涂色结果。（不能留有空白格子）
我们考虑怎样的结果序列是合法的，显然必须有
一段长度为k的连续区间为同一颜色，因为你最后一下
肯定要刷k个相同。我们可以发现，只要满足有这么一段
就都是合法方案。因此我们就是要统计至少有一段
k个相同的数的序列个数。我们发现求它的补集更好求，
即没有一段k个相同的数的序列个数。那么答案就是
总方案数m^n-补集方案数。（我就是这点没想到，gg了！）
那么这个就比较好求了，我们可以dp。
dp[i]，表示i个格子的方案数，则i< k时，显然有dp[i]=m^i，
i>=k时，我们可以通过枚举最后一段相同颜色的长度j（显然是1~k-1的）来转移即可，有
$dp[i]=\sum_{j=1}^{k-1}dp[i-j]*(m-1)$。
然后后缀和优化一下就是O(n)的。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 1000010
#define ll long long
#define inf 0x3f3f3f3f
#define mod 1000000007
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,k,dp[N],sum=0;
inline int ksm(int x,int k){
    int res=1;
    for(;k;k>>=1,x=(ll)x*x%mod)
        if(k&1) res=(ll)res*x%mod;return res;
}
int main(){
    n=read();m=read();k=read();dp[0]=1;
    for(int i=1;i<k;++i) dp[i]=(ll)dp[i-1]*m%mod,sum=(sum+dp[i])%mod;
    for(int i=k;i<=n;++i){
        dp[i]=(ll)sum*(m-1)%mod;sum+=dp[i]-dp[i-k+1];
        if(sum<0) sum+=mod;sum%=mod;
    }int ans=ksm(m,n)-dp[n];if(ans<0) ans+=mod;
    printf("%d\n",ans);
    return 0;
}