【LOJ #6074】「2017 山东一轮集训 Day6」子序列(矩阵乘法)

传送门


好像pkuwc2020Day1T1pkuwc2020Day1T1处理方法就是这个
可惜当时劳资没做过这题。。。

显然有一个dpdp是设f[i][j]f[i][j]
表示前ii个最后一个字符是j(j[0,8])j(j\in[0,8])的方案
f[i][9]f[i][9]则表示之前一个都没选的方案
初始f[0][9]=1f[0][9]=1
然后有f[i][s[i]]=k=09f[i1][k],f[i][j]=f[i1][j](js[i])f[i][s[i]]=\sum_{k=0}^9 f[i-1][k],f[i][j]=f[i-1][j](j\not=s[i])
其实就是序列自动机上拓扑排序求路径数
所以正确性显然

发现这个转移可以写成矩阵乘法的形式
每次转移就是乘上一个矩阵
Ai=[1010001100 1000010000101]A_i=\left[\begin{matrix}1&0 \cdots1 &0\cdots0 \\0 & 1 \cdots1 &0 \cdots 0 \\ \vdots &\ddots\ 1 &0\cdots 0\\0 &0\cdots1 &0\cdots 0\\0 & 0 \cdots 1 &0 \cdots 1 \end{matrix}\right]
就是A[i][i]=1,A[i][s[i]]=1A[i][i]=1,A[i][s[i]]=1

实际答案就是
[001]AlAl+1Ar[111]\left[\begin{matrix} 0&0&\cdots &1\end{matrix}\right]*A_l*A_{l+1}\cdots*A_{r}*\left[\begin{matrix} 1\\ 1\\ \vdots \\1\end{matrix}\right]
考虑预处理AA的前缀积
B=A1B=A^{-1}
发现BB就是
[1010001100 1000010000101]\left[\begin{matrix}1&0 \cdots -1 &0\cdots0 \\0 & 1 \cdots -1 &0 \cdots 0 \\ \vdots &\ddots\ 1 &0\cdots 0\\0 &0\cdots-1 &0\cdots 0\\0 & 0 \cdots -1 &0 \cdots 1 \end{matrix}\right]
AB=BA=IA*B=B*A=I
就是B[i][i]=1,B[i][s[i]]=1(is[i])B[i][i]=1,B[i][s[i]]=-1(i\not =s[i])
那答案就是
[001]Bl1Bl2B1AlAl+1Ar[111]\left[\begin{matrix} 0&0&\cdots &1\end{matrix}\right]*B_{l-1}*B_{l-2}\cdots*B_1*A_l*A_{l+1}\cdots*A_{r}*\left[\begin{matrix} 1\\ 1\\ \vdots \\1\end{matrix}\right]

于是考虑预处理BB左乘的前缀积,AA右乘的前缀积即可

直接做是(n+q)m3(n+q)m^3的,m=10m=10

考虑怎么快速处理A,BA,B
右乘一个AiA_i就是把每一行第s[i]s[i]个变成这一行的和
于是直接对每一行维护一个和即可
最后乘一个[111]\left[\begin{matrix} 1\\ 1\\ \vdots \\1\end{matrix}\right]实际上就是要每一行的和

左乘一个BiB_i就是对于每一列,除了s[i]s[i]行,其他每个位置都减去第s[i]s[i]行的值
于是维护一个减的标记即可
若原来一列是[a1x,a2x,a3x][a_1-x,a_2-x,a_3-x]
之后就是[a1a2,(2a2x)a2,a3a2][a_1-a_2,(2*a_2-x)-a_2,a_3-a_2]
直接维护就好了
最后左乘一个[001]\left[\begin{matrix} 0&0&\cdots &1\end{matrix}\right]就是要最后一行的所有数

于是可以在O((n+q)m)O((n+q)m)做完了

#include<bits/stdc++.h>
using namespace std;
#define cs const
#define re register
#define pb push_back
#define pii pair<int,int>
#define ll long long
#define fi first
#define se second
#define bg begin
cs int RLEN=1<<20|1;
inline char gc(){
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob)&&(ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob)?EOF:*ib++;
}
inline int read(){
    char ch=gc();
    int res=0;bool f=1;
    while(!isdigit(ch))f^=ch=='-',ch=gc();
    while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=gc();
    return f?res:-res;
}
inline int readstr(int *s){
	int top=0;char ch=gc();
	while(isspace(ch))ch=gc();
	while(!isspace(ch))s[++top]=ch-'a',ch=gc();
	return top;
}
template<class tp>inline void chemx(tp &a,tp b){a<b?a=b:0;}
template<class tp>inline void chemn(tp &a,tp b){a>b?a=b:0;}
cs int mod=1e9+7;
inline int add(int a,int b){return (a+=b)>=mod?(a-mod):a;}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){static ll r;r=1ll*a*b;return (r>=mod)?(r%mod):r;}
inline void Add(int &a,int b){(a+=b)>=mod?(a-=mod):0;}
inline void Dec(int &a,int b){a-=b,a+=a>>31&mod;}
inline void Mul(int &a,int b){static ll r;r=1ll*a*b;a=(r>=mod)?(r%mod):r;}
inline int ksm(int a,int b,int res=1){for(;b;b>>=1,Mul(a,a))(b&1)&&(Mul(res,a),1);return res;}
inline int Inv(int x){return ksm(x,mod-2);}
inline int fix(int x){return (x<0)?x+mod:x;}
cs int N=100005;
int a[N][10],b[N][10],ta[10][10],sum[10],tb[10][10],tag[10],n;
int s[N];
int main(){
	#ifdef Stargazer
	freopen("lx.in","r",stdin);
	#endif
	n=readstr(s);
	b[0][9]=1;
	for(int i=0;i<10;i++)tb[i][i]=1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<10;j++){
			int k=tag[j];tag[j]=tb[s[i]][j];
			tb[s[i]][j]=dec(mul(tb[s[i]][j],2),k);
		}
		for(int j=0;j<10;j++)b[i][j]=dec(tb[9][j],tag[j]);
	}
	for(int i=0;i<10;i++)sum[i]=a[i][i]=1,ta[i][i]=1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<10;j++){
			int tp=sum[j];Dec(sum[j],ta[j][s[i]]);
			ta[j][s[i]]=tp,Add(sum[j],ta[j][s[i]]);
		}
		for(int j=0;j<10;j++)a[i][j]=sum[j];
	}
	int q=read();
	while(q--){
		int l=read(),r=read();
		int res=0;
		for(int i=0;i<10;i++)Add(res,mul(b[l-1][i],a[r][i]));
		cout<<dec(res,1)<<'\n';
	}
}
posted @ 2020-02-18 20:26  Stargazer_cykoi  阅读(266)  评论(0编辑  收藏  举报