Codeforces 1149 B - Three Religions

B - Three Religions

思路：dp

dp[i][j][k]：a的前i个和b的前j个和c的前k个能构成的最前面的位置

删字符时状态不用改变，加字符时只会改变1*250*250个状态

代码：

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
#include<bits/stdc++.h>
using namespace std;
#define y1 y11
#define fi first
#define se second
#define pi acos(-1.0)
#define LL long long
#define ls rt<<1, l, m
#define rs rt<<1|1, m+1, r
//#define mp make_pair
#define pb push_back
#define ULL unsigned LL
#define pll pair<LL, LL>
#define pli pair<LL, int>
#define pii pair<int, int>
#define piii pair<pii, int>
#define pdi pair<double, int>
#define pdd pair<double, double>
#define mem(a, b) memset(a, b, sizeof(a))
#define debug(x) cerr << #x << " = " << x << "\n";
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
//head
 
const int N = 1e5 + ;
char s[N];
int nxt[N][], n, q, x;
char op[], cc[];
int dp[][][];
string a, b, c;
int main() {
    scanf("%d %d", &n, &q);
    scanf("%s", s+);
    for (int i = ; i < ; ++i) nxt[n+][i] = n+, nxt[n+][i] = n+;
    for (int i = n; i >= ; --i) {
        for (int j = ; j < ; ++j) {
            if(j == s[i]-'a') nxt[i][j] = i;
            else nxt[i][j] = nxt[i+][j];
        }
    }
    while(q--) {
        scanf("%s %d", op, &x);
        if(op[] == '+') {
            scanf("%s", cc);
            if(x == ){
                a.push_back(cc[]);
                for (int i = ; i <= b.size(); ++i) {
                    for (int j = ; j <= c.size(); ++j) {
                        dp[a.size()][i][j] = n+;
                        if(a.size() > ) dp[a.size()][i][j] = min(dp[a.size()][i][j], nxt[dp[a.size()-][i][j]+][a.back()-'a']);
                        if(i > ) dp[a.size()][i][j] = min(dp[a.size()][i][j], nxt[dp[a.size()][i-][j]+][b[i-]-'a']);
                        if(j > ) dp[a.size()][i][j] = min(dp[a.size()][i][j], nxt[dp[a.size()][i][j-]+][c[j-]-'a']);
                    }
                }
            }
            else if(x == ) {
                b.push_back(cc[]);
                for (int i = ; i <= a.size(); ++i) {
                    for (int j = ; j <= c.size(); ++j) {
                        dp[i][b.size()][j] = n+;
                        if(b.size() > ) dp[i][b.size()][j] = min(dp[i][b.size()][j], nxt[dp[i][b.size()-][j]+][b.back()-'a']);
                        if(i > ) dp[i][b.size()][j] = min(dp[i][b.size()][j], nxt[dp[i-][b.size()][j]+][a[i-]-'a']);
                        if(j > ) dp[i][b.size()][j] = min(dp[i][b.size()][j], nxt[dp[i][b.size()][j-]+][c[j-]-'a']);
                    }
                }
            }
            else if(x == ) {
                c.push_back(cc[]);
                for (int i = ; i <= a.size(); ++i) {
                    for (int j = ; j <= b.size(); ++j) {
                        dp[i][j][c.size()] = n+;
                        if(c.size() > ) dp[i][j][c.size()] = min(dp[i][j][c.size()], nxt[dp[i][j][c.size()-]+][c.back()-'a']);
                        if(i > ) dp[i][j][c.size()] = min(dp[i][j][c.size()], nxt[dp[i-][j][c.size()]+][a[i-]-'a']);
                        if(j > ) dp[i][j][c.size()] = min(dp[i][j][c.size()], nxt[dp[i][j-][c.size()]+][b[j-]-'a']);
                    }
                }
            }
        }
        else {
            if(x == ) a.pop_back();
            else if(x == ) b.pop_back();
            else if(x == ) c.pop_back();
        }
        if(dp[a.size()][b.size()][c.size()] > n) printf("NO\n");
        else printf("YES\n");
    }
    return ;
}