CF750E New Year and Old Subsequence题解——dp既出，矩阵归来

Description

给定一个字符串，每个字符都是$0-9$之间的数。

每次查询在一段区间$[l,r]$中至少需要删掉多少个字符，才能使得剩下的字符串不包含子序列2016且包含子序列2017。

Solution

考虑$dp$。

$d{p}_{i,j}:$ 目前匹配到了串2017的第$j$位的最少删去字符数量。

状态转移如下:

①若$a_i=2$。
(1)不删去$a_i$。此时$a_i$的作用在于将一个暂未匹配的状态($\varnothing$)变为一个匹配$1$位的状态(“2”)。在这种决策下有$d{p}_{i,1}=d{p}_{i-1,0}$。
(2)删去$a_i$。可以发现删除$a_i$起到了保留状态的作用，可以不改变一个匹配$0$位的状态。在这种决策下有$d{p}_{i,0}=d{p}_{i-1,0}+1$。

②若$a_i=0$，与①同理:
(1)$d{p}_{i,2}$的决策点包含$d{p}_{i-1,1}$；
(2)$d{p}_{i,1}$的决策点包含$d{p}_{i-1,1}+1$。

③若$a_i=1/7$，与①②同理，请自行推导。
④若$a_i=6$：
(1)不删去$a_i$。这种转移合法当且仅当当前匹配到的位数不超过$2$，否则转移不合法。为什么这里有一个状态转移的合法限制呢？假设匹配了$3$位，那么此时就含有子序列$201$，添上一个$6$后就存在了子序列$2016$而这时题目严格禁止出现的。匹配了$4$位的情况同理。所以，$d{p}_{i,0},d{p}_{i,1},d{p}_{i,2}$可以直接转移过来，但$d{p}_{i,3},d{p}_{i,4}$必须分别从$d{p}_{i-1,3},d{p}_{i-1,4}$加$1$而转移过来。

⑤若$a_i$为其他的数，那么不会有任何影响，直接转移过来即可。

对于每次询问，扫一遍整个区间进行转移，时间复杂度为$O(5^{2}qn)$，无法通过。

---

这一类含有“区间查询”且不含区间修改的$dp$是经典的动态$dp$，即线段树维护矩乘的模型。

这句话说得有亿点绕？没关系，我们形象地解释一遍。

不难发现当前的状态是一个长度为$5$的小数组，每经过一个位置就要经过对应的变换(转移)。这种变换可以转化为广义矩阵乘法的形式。

更具体地说，状态转移形如$d{p}_{i,j}=d{p}_{i-1,p}+cost(p,j)$，我们可以大胆地重定义矩阵乘法为$$C_{i,j}=\underset{k=1}{\overset{n}{\min }}(A_{i,k}+B_{k,j})$$

显然这是有结合律的。对于每一个位置的对应状态转移可以压缩为一个矩阵，表示“当前”状态通过这个位置要乘上这个矩阵。

最后我们分析每次区间查询的本质: 等价于一次区间查询矩阵乘积。我们可以采用线段树维护区间矩乘，从而解决了本题。

时间复杂度为$O(5^{3}n\log n)$。

Code

#include <bits/stdc++.h>
#define inf 2000000007
#define int long long
using namespace std;
const int maxl=200005;
int read(){
	int s=0,w=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-')  w=-w;ch=getchar();}
	while (ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+(ch^'0');ch=getchar();}
	return s*w;
} 

int n,q,le,ri;
int s[maxl];

struct Matrix{
	int a[5][5];
}tree[maxl*4],b[maxl];

Matrix operator * (const Matrix &x,const Matrix &y){
	Matrix z;
	for (int i=0;i<5;i++){
		for (int j=0;j<5;j++){
			z.a[i][j]=inf;
			for (int k=0;k<5;k++)
			  z.a[i][j]=min(z.a[i][j],x.a[i][k]+y.a[k][j]);
		}
	}
	return z;
}

void pushup(int rt){tree[rt]=tree[2*rt]*tree[2*rt+1];};
void build_tree(int l,int r,int rt){
	if (l==r){
		tree[rt]=b[l];
		return;
	}
	int mid=(l+r)>>1;
	build_tree(l,mid,2*rt),build_tree(mid+1,r,2*rt+1);
	pushup(rt);
}

Matrix query(int nl,int nr,int l,int r,int rt){
	if (nl<=l&&r<=nr){
		return tree[rt];
	}
	int mid=(l+r)>>1;Matrix res;
	if (nl<=mid&&nr<=mid)  res=query(nl,nr,l,mid,2*rt);
	else if (nl>mid&&nr>mid)  res=query(nl,nr,mid+1,r,2*rt+1);
	else if (nl<=mid&&nr>mid)  res=query(nl,nr,l,mid,2*rt)*query(nl,nr,mid+1,r,2*rt+1);
	return res;
}

signed main(){
	n=read(),q=read();
	for (int i=1;i<=n;i++){
		char tmpxl;
		cin>>tmpxl;
		s[i]=tmpxl-'0';
	}
	for (int i=1;i<=n;i++){
		for (int j=0;j<5;j++){
			for (int k=0;k<5;k++){
				if (j!=k)  b[i].a[j][k]=inf;
				else b[i].a[j][k]=0;
			}
		}
		if (s[i]==2)  b[i].a[0][0]=1,b[i].a[0][1]=0;
		if (s[i]==0)  b[i].a[1][1]=1,b[i].a[1][2]=0;
		if (s[i]==1)  b[i].a[2][2]=1,b[i].a[2][3]=0;
		if (s[i]==7)  b[i].a[3][3]=1,b[i].a[3][4]=0;
		if (s[i]==6)  b[i].a[3][3]=1,b[i].a[4][4]=1;
	}
	build_tree(1,n,1);
	while (q--){
		le=read(),ri=read();
		int ans=query(le,ri,1,n,1).a[0][4];
		if (ans>ri-le+1)  puts("-1");
		else printf("%lld\n",ans);
	}
	return 0;
}