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;
}
发布了455 篇原创文章 · 获赞 16 · 访问量 22万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览