Educational Codeforces Round 56 (Rated for Div. 2) F - Vasya and Array dp好题

F - Vasya and Array

dp[ i ][ j ] 表示用了前 i 个数字并且最后一个数字是 j 的方案数。

dp[ i ][ j ] = sumdp [i - 1 ][ j ]， 这样的话会有不合法的方案算进去，而且不合法的方案只有 i - len + 1 到 i 这一段相同才会

出现， 所以如果 i - len + 1 到 i 可以变成一样的话要减去 sumdp[ i - len ] - dp[ i - len ][ j ]

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PLL pair<LL, LL>
#define PLI pair<LL, int>
#define PII pair<int, int>
#define SZ(x) ((int)x.size())
#define ull unsigned long long
#define x first
#define y second
using namespace std;

const int N = 1e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = ;
const double eps = 1e-;

int n, k, len, a[N];
int dp[N][], sum[N], cnt[N][];

inline void add(int &a, int b) {
    a += b; if(a >= mod) a -= mod;
}

int main() {
    scanf("%d%d%d", &n, &k, &len);
    for(int i = ; i <= n; i++) scanf("%d", &a[i]);
    for(int i = ; i <= n; i++)
        for(int j = ; j <= k; j++)
            cnt[i][j] = cnt[i-][j] + (a[i]==j || a[i]==-);
    sum[] = ;
    for(int i = ; i <= n; i++) {

        if(i < len) {
            if(~a[i]) {
                add(dp[i][a[i]], sum[i-]);
            } else {
                for(int j = ; j <= k; j++)
                    add(dp[i][j], sum[i-]);
            }
        } else {
            if(~a[i]) {
                add(dp[i][a[i]], sum[i-]);
                if(cnt[i][a[i]]-cnt[i-len][a[i]] == len) {
                    add(dp[i][a[i]], mod-(sum[i-len]-dp[i-len][a[i]]+mod)%mod);
                }

            } else {
                for(int j = ; j <= k; j++) {
                    add(dp[i][j], sum[i-]);
                    if(cnt[i][j]-cnt[i-len][j]==len) {
                        add(dp[i][j], mod-(sum[i-len]-dp[i-len][j]+mod)%mod);
                    }
                }
            }
        }
        for(int j = ; j <= k; j++)
            add(sum[i], dp[i][j]);
    }
    printf("%d\n", sum[n]);
    return ;
}

/*
*/