2023“钉耙编程”中国大学生算法设计超级联赛（2）Hello World 3 Pro Max

2023“钉耙编程”中国大学生算法设计超级联赛（2）Hello World 3 Pro Max

题目大意

有一个长度为$n$的字符串$S$，有小写字母$h,e,l,o,w,r,d$组成。对于$S$中的每个字符，这个字符为$h$的概率为$p_1$，为$e$的概率为$p_2$，以此类推。输入中有$7$个数$P_1$到$P_7$，令$sum=P_1+P_2+\cdots +P_7$，则$p_i=\frac{P_i}{sum}$。

一开始，字符串$S$的每个位置都是未知的。接下来，有$q$组操作，操作有两种：

• 1 r c，表示确定了第$x$个字符为$c$
• 2 l r，表示求$S$的子串$S(l,r)$中与$helloword$相同的子序列的期望数量对$10^9+7$取模后的值

有$t$组数据。

$1\le t\le 10,1\le n\le 5\times 10^4,1\le P_i\le 10^8,1\le q\le 5\times 10^4$

数据保证$\sum n\le 5\times 10^4,\sum q\le 5\times 10^4$

题解

考虑$DP$，设$f_{i,j}$表示$S$的子串$S(1,i)$中匹配到目标串的第$j$位的子序列数的期望数量，则

• 当$S_i$确定时，$f_{i,j}=f_{i-1,j}+(S_i==T_j)\cdot f_{i-1,j-1}$
• 当$S_i$不确定时，$f_{i,j}=f_{i-1,j}+p_{T_j}\cdot f_{i-1,j-1}$

这样做的话，单次询问的时间复杂度为$O(11n)$，总时间复杂度为$O(11nq)$，过不了。

我们考虑将$DP$转化为矩阵。$f_i$是$11$维的列向量，大小为$11\times 11$的矩阵$A_i$表示转移关系，$f_i=f_{i-1}\times A_i$。

在每一个$A$中，$A_{j,j}=1$，

• 当$S_i$确定时，$A_{j-1,j}=(S_i==T_j)$
• 当$S_i$不确定时，$A_{j-1,j}=p_{T_j}$

其余部分都为$0$。

用线段树来维护矩阵，每次询问就是单点修改和区间查询。总时间复杂度为$O(11^3(n+q\log n))$。

这样可能还是会$\text{TLE}$，我们可以在加一些小小的优化。

因为上述的矩阵都是上三角矩阵，所以在矩阵相乘的时候可以将$(A\times B{)}_{i,j}=\sum _{k=0}^{11}{A}_{i,k}\times B_{k,j}$优化为$(A\times B{)}_{i,j}=\sum _{k=i}^j{A}_{i,k}\times B_{k,j}$。

这样就能将前面的$11^3$优化到两百多，是可以通过这题的。

code

#include<bits/stdc++.h>
#define lc k<<1
#define rc k<<1|1
using namespace std;
const long long mod=1000000007;
int t,n,q,v[15];
char wt[15];
long long sum,ny,p[15];
struct matrix{
	long long a[15][15];
	matrix operator*(const matrix ax)const{
		matrix cx;
		for(int i=0;i<11;i++){
			for(int j=0;j<11;j++){
				cx.a[i][j]=0;
				for(int k=i;k<=j;k++){
					cx.a[i][j]=(cx.a[i][j]+a[i][k]*ax.a[k][j]%mod)%mod;
				}
			}
		}
		return cx;
	}
}w,ans,tr[500005];
void init(){
	v[1]=1;
	v[2]=2;
	v[3]=v[4]=v[9]=3;
	v[5]=v[7]=4;
	v[6]=5;
	v[8]=6;
	v[10]=7;
	
	wt[1]='h';
	wt[2]='e';
	wt[3]=wt[4]=wt[9]='l';
	wt[5]=wt[7]='o';
	wt[6]='w';
	wt[8]='r';
	wt[10]='d';
}
long long mi(long long tt,long long vv){
	if(!vv) return 1;
	long long re=mi(tt,vv/2);
	re=re*re%mod;
	if(vv&1) re=re*tt%mod;
	return re;
}
void to(matrix &tw){
	for(int i=0;i<11;i++){
		for(int j=0;j<11;j++){
			tw.a[i][j]=0;
		}
		tw.a[i][i]=1;
	}
}
void build(int k,int l,int r){
	if(l==r){
		tr[k]=w;
		return;
	}
	int mid=l+r>>1;
	build(lc,l,mid);build(rc,mid+1,r);
	tr[k]=tr[lc]*tr[rc];
}
void ch(int k,int l,int r,int x,char c){
	if(l==x&&l==r){
		for(int i=0;i<11;i++){
			tr[k].a[i-1][i]=(c==wt[i]);
		}
		return;
	}
	int mid=l+r>>1;
	if(x<=mid) ch(lc,l,mid,x,c);
	else ch(rc,mid+1,r,x,c);
	tr[k]=tr[lc]*tr[rc];
}
void find(int k,int l,int r,int x,int y){
	if(l>=x&&r<=y){
		ans=ans*tr[k];
		return;
	}
	int mid=l+r>>1;
	if(x<=mid) find(lc,l,mid,x,y);
	if(y>mid) find(rc,mid+1,r,x,y);
}
int main()
{
	init();
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		sum=0;
		for(int i=1;i<=7;i++){
			scanf("%lld",&p[i]);
			sum+=p[i];
		}
		ny=mi(sum,mod-2);
		for(int i=1;i<=7;i++){
			p[i]=p[i]*ny%mod;
		}
		to(w);
		for(int i=1;i<11;i++){
			w.a[i-1][i]=p[v[i]];
		}
		build(1,1,n);
		scanf("%d",&q);
		int tp,x,y;
		char c;
		while(q--){
			scanf("%d",&tp);
			if(tp==1){
				scanf("%d",&x);
				c=getchar();
				while(c<'a'||c>'z') c=getchar();
				ch(1,1,n,x,c);
			}
			else{
				scanf("%d%d",&x,&y);
				to(ans);
				find(1,1,n,x,y);
				printf("%lld\n",ans.a[0][10]);
			}
		}
	}
	return 0;
}