bzoj 5190: [Usaco2018 Jan]Stamp Painting

题意：

给出$n,m,k$问由m种颜色组成长度为n的序列，且至少有长度$>=k$的一段颜色相同的方案数。

题解：

无脑暴力容斥非正解
枚举有多少段颜色，然后容斥即可：

$$ans=\sum_{i=1}^n\sum_{j=1}^i(-1)^{j+1}C_{n-kj-1}^{i-1}*C_i^j*m*(m-1)^{i-1}$$
code：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#define LL long long
using namespace std;
const LL mod=1e9+7;
LL f[1000010],inv[1000010];
void pre()
{
    f[0]=f[1]=inv[0]=inv[1]=1;
    for(int i=2;i<=1000000;i++) f[i]=f[i-1]*i%mod,inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    for(int i=2;i<=1000000;i++) inv[i]=inv[i-1]*inv[i]%mod;
}
LL C(int n,int m)
{
    if(n>m) return 0;
    return f[m]*inv[m-n]%mod*inv[n]%mod;
}
int n,m,k;
LL pow[1000010];
int main()
{
    pre();
    scanf("%d %d %d",&n,&m,&k);k--;
    LL ans=0;
    pow[0]=1;for(int i=1;i<=n;i++) pow[i]=pow[i-1]*(m-1)%mod;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++)
        {
            if(n-k*j-1<i-1) break;
            ans=(ans+((j&1)?1:-1)*C(i-1,n-k*j-1)*C(j,i)%mod*pow[i-1]%mod)%mod;(ans+=mod)%=mod;
        }
    printf("%lld",m*ans%mod);
}