CodeForces - 1093F：Vasya and Array （DP&计数）

题意：N,K,L，以及给定长度为N的序列，表示其对应的颜色，-1表示还没有涂色，现在让你去涂色，使得最后没有大于等于L的连续的同色的情况。

思路：我们用dp[i][j]表示第i个位置颜色为j的合法方案数，用sum[i]表示dp[i][1]+dp[i][2]+...dp[i][k]。

那么a[i]==j或者-1时，dp[i][j]=sum[i-1]-x。然后我们考虑到可能有不合法的情况x，当且仅当前面有长度为L的序列颜色为j或者-1时，其数量为sum[i-L]-sum[i-L][j]，减去后面这个sum[i-L][j]是因为它长度为L+1了，那么之前已经减去了。

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int Mod=;
const int maxn=;
int a[maxn],dp[maxn][],sum[maxn],len[];
int main()
{
    int N,K,L;
    scanf("%d%d%d",&N,&K,&L);
    rep(i,,N) scanf("%d",&a[i]);
    sum[]=;
    rep(i,,N){
        rep(j,,K) len[j]=(a[i]==-||a[i]==j)?len[j]+:;
        if(a[i]==-){
            rep(j,,K) {
                dp[i][j]=sum[i-];
                if(len[j]>=L) dp[i][j]=(dp[i][j]-(sum[i-L]-dp[i-L][j]+Mod)%Mod+Mod)%Mod;
            }
        }
        else  {
            dp[i][a[i]]=sum[i-];
            if(len[a[i]]>=L) dp[i][a[i]]=(dp[i][a[i]]-(sum[i-L]-dp[i-L][a[i]]+Mod)%Mod+Mod)%Mod;
        }
        rep(j,,K) sum[i]=(sum[i]+dp[i][j])%Mod;
    }
    printf("%d\n",sum[N]);
    return ;
}