• LOJ.6074.[2017山东一轮集训Day6]子序列(DP 矩阵乘法)


    题目链接

    参考yww的题解。本来不想写来但是他有一些笔误...而且有些地方不太一样就写篇好了。
    不知不觉怎么写了这么多...
    另外还是有莫队做法的...(虽然可能卡不过)


    (60)分的(O(n^2))做法就是,令(f[i])表示以(s[i])结尾的不同子序列个数,(las[c])表示(c)字符上次出现的位置(没有出现过则为(-1)),转移是:$$f[i]=egin{cases}2f[i-1]+1&,las[s[i]]=-12f[i-1]-f[las[s[i]]]&,las[s[i]] eq-1end{cases}$$


    上面这个做法挺妙的,但是好像没什么优化空间。
    考虑另一种DP,令(f[i][j])表示前(i)个字符,以字符(j)结尾的不同子序列个数。转移为:$$f[i][j]=egin{cases}f[i-1][j-1]&,j eq s_i\sum_{k=0}^mf[i-1][k]&,j=s_iend{cases}$$

    其中字符集是(0sim m-1)(m=9)),特别的令(f[i][m])表示前(i)个字符什么也没选的方案数,因为子序列可以从任意位置开始。
    这样不会算重,感觉也挺妙的。。(好吧是自己菜)

    显然可以用矩阵把转移表示出来。令$$F_i=left[egin{matrix}f_{i,1}f_{i,2}\vdotsf_{i,m+1}end{matrix} ight]$$

    转移矩阵就是(A[i][i]=1,A[c][j]=1 (c=s[i]))(把单位矩阵(s[i])那一行全设为(1))。就是这样子:$$F_i=A_iF_{i-1}A_i=left[egin{matrix}1&&&&&1&&&1&1&1&1&1&&&1&&&&&1end{matrix} ight]$$

    再考虑一下初始化、最后的求和,令$$egin{aligned}U&=left[egin{matrix}11\vdots1end{matrix} ight]V&=left[egin{matrix}0 0 cdots 0 1end{matrix} ight]end{aligned}$$

    (A_i)显然有逆矩阵(可以比较容易地写出来)。那么区间([l,r])的答案就是$$egin{aligned}&UA_rA_{r-1}cdots A_lV=&UA_rA_{r-1}cdots A_1{A_1}^{-1}{A_2}^{-1}ldots A_{l-1}Vend{aligned}$$

    这里把(U)放到了前面,(V)放到了后面,都一样,我觉得还是这样方便一些...
    需要注意矩阵乘法没有交换律,注意乘的顺序。
    所以预处理一个转移矩阵的前缀积(f_i)、转移矩阵逆元的前缀积(g_i),就可以(O(m^3))回答一次询问了。
    预处理(f_i)的时候让它和(U)乘一下,同理(g_i)(V)乘一下,询问就是(O(m))的了。

    但是预处理的复杂度还是(O(nm^3))的(但是开O2已经能过了...)。
    注意到转移矩阵非常特殊,一个矩阵(M)乘上(A_i)时,(M_{s_i,j}')(A_i)(j)列元素的和,(M')其它行的元素不变。这样乘(A_i)可以做到(O(m))
    同时左乘(U)得到的矩阵就是对列求和。
    那么我们维护(f_i)的时候(乘了(U),是个(1 imes n)的),第(j)列的和即(f_{i,j}),就是上一次第(j)列的和(*2)减去(A_{s_i,j}),上一次第(j)列的和就是(f_{i-1,j})。那么这个转移也是(O(m))的。

    同理,(A_i^{-1})大概是这样:$$A_i=left[egin{matrix}1&&&&&1&&&-1&-1&1&-1&-1&&&1&&&&&1end{matrix} ight]$$

    (A[i][i]=1,A[c][j]=-1 (c=s[i],j eq i))(把单位矩阵(s[i])那一行除了(A_{s[i],s[i]})全设为(-1))。
    一个矩阵乘(A_i^{-1})时,除了(c=s[i])列之外的列(M_{i,j}),都会减去(M_{i,c}),第(c)列的元素不变。维护一个整行减了多少的标记,对(M_{i,c})单点修改一下即可。
    注意到假设其中一行是:(left[a_1-vquad a_2-vquad a_3-vquad a_4-v ight])(s[i]=3)时,会变成(left[a_1-a_3quad a_2-a_3quad (2a_3-v)-a_3quad a_4-a_3 ight]),也就是对(3)单独修改一下,所有数每次会减掉上次那个数,打个标记修改也是(O(m))的了。

    维护(g_i)时,注意到右乘一个(V)就是把矩阵最后一列取出来,直接求即可。同样(O(m))

    那么总复杂度就做到(O((n+q)m))啦。


    //263ms	8760K
    #include <cstdio>
    #include <cctype>
    #include <cstring>
    #include <algorithm>
    #define rg register
    #define mod 1000000007
    #define Mod(x) x>=mod&&(x-=mod)
    #define Add(x,v) (x+=v)>=mod&&(x-=mod)
    #define Add2(x,y) (x+y>=mod?x+y-mod:x+y)
    //#define gc() getchar()
    #define MAXIN 500000
    #define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
    typedef long long LL;
    const int N=1e5+7,M=10;
    const LL LIM=6e18;
    
    int s[N],f[N][M],g[N][M];
    char IN[MAXIN],*SS=IN,*TT=IN;
    
    inline int read()
    {
    	int now=0;register char c=gc();
    	for(;!isdigit(c);c=gc());
    	for(;isdigit(c);now=now*10+c-48,c=gc());
    	return now;
    }
    void Pre(const int n)
    {
    	static int A[M][M],B[M][M],tag[M];
    	for(int i=0; i<M; ++i) A[i][i]=B[i][i]=f[0][i]=1;
    	g[0][M-1]=1;
    	for(rg int i=1; i<=n; ++i)
    	{
    		rg int c=s[i];
    		for(rg int j=0; j<M; ++j)
    		{
    			rg int tmp=f[i-1][j]<<1; Mod(tmp);
    			f[i][j]=Add2(tmp,mod-A[c][j]);
    			A[c][j]=f[i-1][j];
    			tmp=B[j][c]<<1, Mod(tmp); rg int tmp2=B[j][c];
    			g[i][j]=Add2(B[j][M-1],mod-B[j][c]);
    			B[j][c]=Add2(tmp,mod-tag[j]), tag[j]=tmp2;
    		}
    	}
    }
    
    int main()
    {
    	int n=0;
    	register char c; while(isalpha(c=gc())) s[++n]=c-'a';
    	Pre(n);
    	for(int q=read(); q--; )
    	{
    		int l=read()-1,r=read();
    		LL ans=0;
    		if(l) for(int i=0; i<M; ++i) ans+=1ll*f[r][i]*g[l][i], ans>=LIM&&(ans%=mod);
    		else ans=f[r][M-1];
    		printf("%lld
    ",(ans-1)%mod);
    	}
    
    	return 0;
    }
    
  • 相关阅读:
    codeforce 148D. Bag of mice[概率dp]
    poj2096 Collecting Bugs[期望dp]
    poj3744 Scout YYF I[概率dp+矩阵优化]
    hdu 5318 The Goddess Of The Moon
    hdu5411 CRB and Puzzle[矩阵优化dp]
    poj3734 Blocks[矩阵优化dp or 组合数学]
    1948 NOI 嘉年华
    教主泡嫦娥[有趣的dp状态设计]
    HDU 4455 Substrings[多重dp]
    1296: [SCOI2009]粉刷匠[多重dp]
  • 原文地址:https://www.cnblogs.com/SovietPower/p/10620539.html
Copyright © 2020-2023  润新知