luogu4187

P4187 [USACO18JAN]Stamp Painting

样例

input
3 2 2
output
6

input
6 10 5
output
190

sol：首先可以发现，对于合法的序列，只要有一串至少连续K个相同的就可以了，其他没有限制

这当然是可以dp辣

dp[i][j]表示前i位没有,当前有j个连续相同,前面没有出现连续K个相同

统计答案的时候就是∑i={K,n} dp[i][K]*Ksm(m,n-i)

转移挺容易的

dp[1][1]=m

dp[i][1]=(m-1)*∑j={1,K-1} dp[i-1][j]

dp[i][j=(2~K)] = dp[i-1][j-1]

但是就这样裸的暴力肯定是n²的

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read()
{
    ll s=;
    bool f=;
    char ch=' ';
    while(!isdigit(ch))
    {
        f|=(ch=='-'); ch=getchar();
    }
    while(isdigit(ch))
    {
        s=(s<<)+(s<<)+(ch^); ch=getchar();
    }
    return (f)?(-s):(s);
}
#define R(x) x=read()
inline void write(ll x)
{
    if(x<)
    {
        putchar('-'); x=-x;
    }
    if(x<)
    {
        putchar(x+'');    return;
    }
    write(x/);
    putchar((x%)+'');
    return;
}
#define W(x) write(x),putchar(' ')
#define Wl(x) write(x),putchar('\n')
const ll Mod=;
const int N=;
int n,m,K;
ll dp[N][N]; //dp[i]表示前i位没有,当前有j个连续相同,前面没有出现连续K个相同
inline void Ad(ll &x,ll y)
{
    x+=y;
    x-=(x>=Mod)?Mod:;
}
inline ll Ksm(ll x,ll y)
{
    ll ans=;
    while(y)
    {
        if(y&) ans=ans*x%Mod;
        x=x*x%Mod;
        y>>=;
    }
    return ans;
}
int main()
{
    freopen("data.in","r",stdin);
    freopen("baoli.out","w",stdout);
    int i,j;
    ll ans=;
    R(n); R(m); R(K);
    dp[][]=m;
    for(i=;i<=n;i++)
    {
        for(j=;j<=(K-)&&j<=(i-);j++) Ad(dp[i][],dp[i-][j]*(m-)%Mod);
        for(j=;j<=K&&j<=i;j++)
        {
            dp[i][j]=dp[i-][j-];
        }
    }
//    for(i=1;i<=n;i++)
//    {
//        for(j=1;j<=K;j++) W(dp[i][j]);
//        puts("");
//    }
    for(i=K;i<=n;i++) Ad(ans,dp[i][K]*Ksm(m,n-i)%Mod);
    Wl(ans);
    return ;
}
/*
input
3 2 2
output
6
*/

然后面临的问题就是怎么优化这个dp，容易发现其实每次除了第一位，另外都是不变的（向右移一位而已），所以只要更新第一位的值可以了

有这样两个队列

1 2 3 4 5              Head=10 Tail=14
             x 1 2 3 4                 Head=9 Tail=13

发现了吗，只要搞一个队列，每次Head-1，Tail-1就会向左移一位，那个红色的x就是要更新的值了

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline ll read()
{
    ll s=;
    bool f=;
    char ch=' ';
    while(!isdigit(ch))
    {
        f|=(ch=='-'); ch=getchar();
    }
    while(isdigit(ch))
    {
        s=(s<<)+(s<<)+(ch^); ch=getchar();
    }
    return (f)?(-s):(s);
}
#define R(x) x=read()
inline void write(ll x)
{
    if(x<)
    {
        putchar('-'); x=-x;
    }
    if(x<)
    {
        putchar(x+'');    return;
    }
    write(x/);
    putchar((x%)+'');
    return;
}
#define W(x) write(x),putchar(' ')
#define Wl(x) write(x),putchar('\n')
const ll Mod=;
const int N=;
int n,m,K;
ll dp[N]; //dp[j]表示前i位(已经滚掉)没有,当前有j个连续相同,前面没有出现连续K个相同
ll Queue[N<<];
ll Bin[N];
inline void Ad(ll &x,ll y)
{
    x+=y;
    x-=(x>=Mod)?Mod:;
    x+=(x<)?Mod:;
}
int main()
{
    freopen("data.in","r",stdin);
    freopen("my.out","w",stdout);
    int i;
    ll Sum=,ans=;
    R(n); R(m); R(K);
    dp[]=m; for(i=;i<=K;i++) dp[i]=;
    Sum=m;
    for(i=;i<=K;i++) Queue[n+i]=dp[i];
    Bin[]=; for(i=;i<=n;i++) Bin[i]=Bin[i-]*m%Mod;
    int Head=n+,Tail=n+K;
    Ad(ans,Queue[Tail]*Bin[n-]%Mod);
    for(i=;i<=n;i++)
    {
        Queue[Head-]=(Sum-Queue[Tail]+Mod)*(m-)%Mod;
        Ad(Sum,Queue[Head-]);
        Ad(Sum,(-)*Queue[Tail]);
        Head--;
        Tail--;
//        for(int j=Head;j<=Tail;j++) W(Queue[j]);
//        puts("");
        Ad(ans,Queue[Tail]*Bin[n-i]%Mod);
    }
    Wl(ans);
    return ;
}
/*
input
3 2 2
output
6

input
6 10 5
output
190
*/
/*
       1 2 3 4 5 Head=10 Tail=14
       1 2 3 4     Head=9 Tail=13
*/

Ps：最后附上对拍

:loop
make.exe
luogu4187.exe
baoli.exe
fc my.out baoli.out
if not errorlevel  goto loop
pause
goto loop

pai

#include <bits/stdc++.h>
using namespace std;
typedef int ll;
inline ll read()
{
    ll s=;
    bool f=;
    char ch=' ';
    while(!isdigit(ch))
    {
        f|=(ch=='-'); ch=getchar();
    }
    while(isdigit(ch))
    {
        s=(s<<)+(s<<)+(ch^); ch=getchar();
    }
    return (f)?(-s):(s);
}
#define R(x) x=read()
inline void write(ll x)
{
    if(x<)
    {
        putchar('-'); x=-x;
    }
    if(x<)
    {
        putchar(x+'');    return;
    }
    write(x/);
    putchar((x%)+'');
    return;
}
#define W(x) write(x),putchar(' ')
#define Wl(x) write(x),putchar('\n')
const int Mod=;
int main()
{
    freopen("data.in","w",stdout);
    srand(time(NULL));
    int n,m,k;
    n=rand()%Mod+;
    m=rand()%Mod+;
    k=rand()%n+;
    W(n); W(m); Wl(k);
    return ;
}

make