2018牛客网暑期ACM多校训练营(第一场)E.Removal

设 next(i, c) 表示位置 i 后第一个字符 c 的位置 f(i, j) 表示当前匹配到 i,删了 j 个,不同的方案数 转移时枚举下一个字符 c,转移到 f(next(i, c), j + next(i, c) - i-1.

最后答案是ans = dp[n-m][0]+dp[n-m+1][1]+......+dp[n][m].

#include"bits/stdc++.h"
using namespace std;
const int MX = 1e5+7;
const int mod = 1e9+7;
typedef long long LL;
int n,m,K;
int a[MX],nxt[MX][11],now[11],dp[MX][11];

void solve()
{
    for(int i = 0; i < 11; i++) now[i] = n+1;
    for(int i = n; i >= 0; i--){
        for(int k = 1; k <= K; k++)
            nxt[i][k] = now[k];
        now[a[i]] = i;
    }

    for(int i = 0; i <= n; i++)
        for(int j = 0; j <= m; j++)
            dp[i][j] = 0;
    dp[0][0] = 1;
    for(int i = 0; i <= n; i++){
        for(int j = 0; j <= m; j++){
            for(int k = 1; k <= K; k++){
                int num = nxt[i][k]-i-1;
                if(j+num <= m) (dp[nxt[i][k]][j+num] += dp[i][j])%=mod;
            }
        }
    }

    int ans = 0;
    for(int j = 0; j <= m; j++)
            (ans += dp[n-j][m-j]) %= mod;
    cout<

 

你可能感兴趣的:(DP)