2019 南昌网络赛C Hello 2019 线段树维护矩阵（dp）

原型题：https://codeforces.com/problemset/problem/750/E

题意：

给你n个数，每次问你l到r区间中至少要去掉几个数使得剩下的数中含有9102序列而不含8102序列

题解：

先把整个串倒过来，这样题目就转化成含2019而不含2018序列

设dp[i][j]为状态i转移到状态j需要删除的最小数，我们用矩阵来表示这个dp

总共5个状态

状态0-null 空

状态1-2

状态2-20

状态3-201

状态4-2019

重载+运算符后用线段树来维护每段区间的矩阵即dp转移方程

特别的，当这个数是8的时候，

e[rt].a[3][3]=1;
e[rt].a[4][4]=1;

201或2019这两个状态不合法，需要删除一个数

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=200050;
const int inf=1e9;
int a[maxn];
struct node{
    int a[5][5];
    node operator + (const node &b) const{
        node c;
        for(int i=0;i<5;i++){
            for(int j=0;j<5;j++){
                c.a[i][j]=inf;
                for(int k=0;k<5;k++)
                    c.a[i][j]=min(c.a[i][j],a[i][k]+b.a[k][j]);
            }
        }
        return c;
    }
}e[maxn<<2];
void build(int l,int r,int rt){
    if(l==r){
        for(int i=0;i<5;i++){
            for(int j=0;j<5;j++)
                e[rt].a[i][j]=(i==j)?0:inf;
        }
        if(a[l]==2){
            e[rt].a[0][0]=1;//0->0 也就是说要删除当前这个2使原来的空串还是空串的花费
            e[rt].a[0][1]=0;//0->1 保留2，无需花费，从空状态转移到“2”状态
        }
        else if(a[l]==0){
            e[rt].a[1][1]=1;
            e[rt].a[1][2]=0;
        }
        else if(a[l]==1){
            e[rt].a[2][2]=1;
            e[rt].a[2][3]=0;
        }
        else if(a[l]==9){
            e[rt].a[3][3]=1;
            e[rt].a[3][4]=0;
        }
        else if(a[l]==8){
            e[rt].a[3][3]=1;
            e[rt].a[4][4]=1;
        }
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    e[rt]=e[rt<<1]+e[rt<<1|1];
}
node query(int l,int r,int ql,int qr,int rt){
    if(l==ql&&r==qr){

        return e[rt];
    }
    int mid=(l+r)/2;
    if(qr<=mid) return query(l,mid,ql,qr,rt<<1);
    else if(ql>mid) return query(mid+1,r,ql,qr,rt<<1|1);
    else return query(l,mid,ql,mid,rt<<1)+query(mid+1,r,mid+1,qr,rt<<1|1);
}
char s[maxn];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=n;i>=1;i--) scanf("%1d",&a[i]);
    build(1,n,1);
    while(m--){
        int l,r;
        scanf("%d%d",&l,&r);
        l=n-l+1;
        r=n-r+1;
        int ans=query(1,n,r,l,1).a[0][4];
        if(ans==inf) printf("-1\n");
        else printf("%d\n",ans);
    }
	return 0;
}