Codeforces problem 750E New Year and Old Subsequence

New Year and Old Subsequence

　　我们的目标是得到含有’2017’而不含有’2016’的子序列，有多次查询，用线段树解决。在线段树的每个节点中，开一个5*5的二维数组arr[i][j]，表示该区间为了能够出现’2017’中的[i,j)子序列而不会出现[i,j]子序列，并且不会出现’2016’，需要删除的最少字符数。
　　比如arr[0][2]表示该区间能够出现’20’但不能出现’201’需要删除的最少字符数。又如arr[1][1]表示不会出现’1’。需要注意的是’6’不能出现在[3][3]（即’1’之后，’7’之前）和[4][4]（即’7’之后）。

#include <bits/stdc++.h>

using namespace std;

#define ll long long

const int INF = 1e9;
const int maxn = 200010;

struct node{
    int val[5][5];
    node(){
        for(int i=0;i<5;i++){
            for(int j=0;j<5;j++){
                val[i][j] = INF;
            }
        }
    }
}tree[maxn<<2];

int n,q;
char num[maxn];
char ch[] = {'2','0','1','7','6'};

int find(char c){
    for(int i=0;i<5;i++){
        if(ch[i]==c)return i;
    }
    return -1;
}

node push_up(node &a,node &b){
    node res;
    for(int i=0;i<5;i++){
        for(int j=i;j<5;j++){
            for(int k=i;k<=j;k++){
                res.val[i][j] = min(res.val[i][j],a.val[i][k]+b.val[k][j]);
            }
        }
    }
    return res;
}

void build_tree(int rt,int l,int r){
    if(l==r){
        int f = find(num[l]);
        for(int i=0;i<5;i++){
            tree[rt].val[i][i] = 0;
        }
        if(f!=-1 && f<4){
            tree[rt].val[f][f+1] = 0;
            tree[rt].val[f][f] = 1;
        }else if(f==4){
            tree[rt].val[3][3] = tree[rt].val[4][4] = 1;
        }
        return;
    }

    int mid = (l+r)>>1;
    build_tree(rt<<1,l,mid);
    build_tree(rt<<1|1,mid+1,r);
    tree[rt] = push_up(tree[rt<<1],tree[rt<<1|1]);
}

node query(int rt,int tl,int tr,int l,int r){
    if(l==tl && r==tr){
        return tree[rt];
    }

    int mid = (tl+tr)>>1;
    if(r<=mid){
        return query(rt<<1,tl,mid,l,r);
    }else{
        if(l>mid){
            return query(rt<<1|1,mid+1,tr,l,r);
        }else{
            node tmpl = query(rt<<1,tl,mid,l,mid);
            node tmpr = query(rt<<1|1,mid+1,tr,mid+1,r);
            return push_up(tmpl,tmpr);
        }
    }
}

int main(){
    cin>>n>>q;
    scanf("%s",num+1);
    build_tree(1,1,n);

    while(q--){
        int l,r;
        scanf("%d %d",&l,&r);
        node ans = query(1,1,n,l,r);
        if(ans.val[0][4] == INF){
            ans.val[0][4] = -1;
        }
        printf("%d\n",ans.val[0][4]);
    } 
    return 0;
}