题目链接
http://acm.hdu.edu.cn/showproblem.php?pid=6155
题解
DP+线代好题。(考场上过多时间刚前两题,没怎么想这题……)
首先列出一个DP式: 设\(dp[i][j]\)表示到第\(i\)位最后一位是\(j\)有多少个本质不同的子序列(最后一位不一定取到第\(i\)位),考虑转移:
假设\(a_i=0\), 那么\(dp[i][0]=2\times dp[i-1][0]+dp[i-1][1]-dp[i-1][0]+1=dp[i-1][0]+dp[i-1][1]+1\), 原因是考虑\(1\)到\(i-1\)中的子序列,可以在后面添一个0
也可以不添,但是添完之后恰好有\(dp[i-1][0]\)个在前面出现过所以减掉,再加上前面以1
结尾的串补上该处的0
和单独一个0
; \(dp[i][1]=dp[i-1][1]\). \(a_i=1\)同理。
(好吧我知道这个DP还有其他的做法,但是这个还是最容易数据结构维护的)
然后考虑如果没有修改怎么维护: 搞一个\(3\times 3\)的矩阵\[\text{A_0}\times \begin{bmatrix}f_0\\f_1\\1\end_{bmatrix}=\begin{bmatrix}f_0+f_1+1\\f_1\\1\end{bmatrix}, \bm\text{A_1}\times \begin{bmatrix}f_0\\f_1\\1\end{bmatrix}=\begin{bmatrix}f_0\\f_0+f_1+1\\1\end{bmatrix\]
很轻易可以得到\[\text{A_0}=\begin{bmatrix}1&1&1\\0&1&0\\0&0&1\end{bmatrix}, \text{A_1}=\begin{bmatrix}1&0&0\\1&1&1\\0&0&1\end{bmatrix}\]
线段树维护区间乘积即可。
区间反转怎么办?维护两棵线段树?可能会被卡常,有更好的方法。(这也是此题的精妙之处)
我们发现矩阵\(\text{A_0}\)经过交换\(1,2\)行、交换\(1,2\)列的操作之后可以变成矩阵\(\text{A_1}\), 矩阵\(\text{A_1}\)经过相同操作也可以变成\(\text{A_0}\).
也就是说我们构造初等矩阵\(\text{E}=\begin{bmatrix}0&1&0\\1&0&0\\0&0&1\end{bmatrix}\),则有\(\text{E}=\text{E}^{-1}\), \(\text{A_1}=\text{E}\text{A_0}\text{E}, \text{A_0}=\text{E}\text{A_1}\text{E}\).
因此有\(\prod^{R}_{i=L}(\text{E}\text{T}_i\text{E})=\text{E}(\prod^R_{i=L}\text{T_i})\text{E}\), 于是直接把乘积矩阵进行上述初等变换即可!
时间复杂度\(O(n\log n)\).
UPD: 刚才发现有大佬用\(2\times 2\)的矩阵维护,大概方法是令\(\bm\text{A_0}\times\begin{bmatrix}f_0+1\\f_1+1\end{bmatrix}=\begin{bmatrix}f_0+f_1+2\\f_1+1\), \(1\)同理。只能说神仙到处是啊……
代码
#include<cstdio> #include<algorithm> #include<cstring> #define llong long long using namespace std; const int N = 1e5; const int P = 1e9+7; void updsum(llong &x,llong y) {x = x+y>=P?x+y-P:x+y;} struct Matrix { llong a[3][3]; Matrix() {a[0][0] = a[0][1] = a[0][2] = a[1][0] = a[1][1] = a[1][2] = a[2][0] = a[2][1] = a[2][2] = 0;} void unitize() {a[0][0] = a[1][1] = a[2][2] = 1ll; a[0][1] = a[0][2] = a[1][0] = a[1][2] = a[2][0] = a[2][1] = 0ll;} Matrix operator *(const Matrix &arg) const { Matrix ret; updsum(ret.a[0][0],a[0][0]*arg.a[0][0]%P); updsum(ret.a[0][0],a[0][1]*arg.a[1][0]%P); updsum(ret.a[0][0],a[0][2]*arg.a[2][0]%P); updsum(ret.a[0][1],a[0][0]*arg.a[0][1]%P); updsum(ret.a[0][1],a[0][1]*arg.a[1][1]%P); updsum(ret.a[0][1],a[0][2]*arg.a[2][1]%P); updsum(ret.a[0][2],a[0][0]*arg.a[0][2]%P); updsum(ret.a[0][2],a[0][1]*arg.a[1][2]%P); updsum(ret.a[0][2],a[0][2]*arg.a[2][2]%P); updsum(ret.a[1][0],a[1][0]*arg.a[0][0]%P); updsum(ret.a[1][0],a[1][1]*arg.a[1][0]%P); updsum(ret.a[1][0],a[1][2]*arg.a[2][0]%P); updsum(ret.a[1][1],a[1][0]*arg.a[0][1]%P); updsum(ret.a[1][1],a[1][1]*arg.a[1][1]%P); updsum(ret.a[1][1],a[1][2]*arg.a[2][1]%P); updsum(ret.a[1][2],a[1][0]*arg.a[0][2]%P); updsum(ret.a[1][2],a[1][1]*arg.a[1][2]%P); updsum(ret.a[1][2],a[1][2]*arg.a[2][2]%P); updsum(ret.a[2][0],a[2][0]*arg.a[0][0]%P); updsum(ret.a[2][0],a[2][1]*arg.a[1][0]%P); updsum(ret.a[2][0],a[2][2]*arg.a[2][0]%P); updsum(ret.a[2][1],a[2][0]*arg.a[0][1]%P); updsum(ret.a[2][1],a[2][1]*arg.a[1][1]%P); updsum(ret.a[2][1],a[2][2]*arg.a[2][1]%P); updsum(ret.a[2][2],a[2][0]*arg.a[0][2]%P); updsum(ret.a[2][2],a[2][1]*arg.a[1][2]%P); updsum(ret.a[2][2],a[2][2]*arg.a[2][2]%P); return ret; } } trans[2]; char a[N+3]; struct SgTNode { Matrix x; bool inv; } sgt[(N<<2)+3]; void build(int u,int le,int ri) { if(le==ri) {sgt[u].x = trans[a[le]]; return;} int mid = (le+ri)>>1; build(u<<1,le,mid); build(u<<1|1,mid+1,ri); sgt[u].x = sgt[u<<1].x*sgt[u<<1|1].x; } void maketag(int u) { sgt[u].inv ^= 1; swap(sgt[u].x.a[0][0],sgt[u].x.a[0][1]); swap(sgt[u].x.a[1][0],sgt[u].x.a[1][1]); swap(sgt[u].x.a[2][0],sgt[u].x.a[2][1]); swap(sgt[u].x.a[0][0],sgt[u].x.a[1][0]); swap(sgt[u].x.a[0][1],sgt[u].x.a[1][1]); swap(sgt[u].x.a[0][2],sgt[u].x.a[1][2]); } void pushdown(int u) { if(sgt[u].inv) { maketag(u<<1); maketag(u<<1|1); sgt[u].inv = 0; } } void inverse(int u,int le,int ri,int lb,int rb) { if(le>=lb && ri<=rb) {maketag(u); return;} pushdown(u); int mid = (le+ri)>>1; if(lb<=mid) {inverse(u<<1,le,mid,lb,rb);} if(rb>mid) {inverse(u<<1|1,mid+1,ri,lb,rb);} sgt[u].x = sgt[u<<1].x*sgt[u<<1|1].x; } Matrix queryprod(int u,int le,int ri,int lb,int rb) { if(le>=lb && ri<=rb) {return sgt[u].x;} pushdown(u); int mid = (le+ri)>>1; Matrix ret; ret.unitize(); if(lb<=mid) {ret = ret*queryprod(u<<1,le,mid,lb,rb);} if(rb>mid) {ret = ret*queryprod(u<<1|1,mid+1,ri,lb,rb);} sgt[u].x = sgt[u<<1].x*sgt[u<<1|1].x; return ret; } int n,q; int main() { int T; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&q); scanf("%s",a+1); for(int i=1; i<=n; i++) a[i] -= 48; trans[0].a[0][0] = 1; trans[0].a[0][1] = 1; trans[0].a[0][2] = 1; trans[0].a[1][1] = 1; trans[0].a[2][2] = 1; trans[1].a[0][0] = 1; trans[1].a[1][0] = 1; trans[1].a[1][1] = 1; trans[1].a[1][2] = 1; trans[1].a[2][2] = 1; build(1,1,n); for(int i=1; i<=q; i++) { int opt,l,r; scanf("%d%d%d",&opt,&l,&r); if(opt==1) { inverse(1,1,n,l,r); } else { Matrix ans = queryprod(1,1,n,l,r); printf("%lld\n",(ans.a[0][2]+ans.a[1][2])%P); } } memset(sgt,0,sizeof(sgt)); } return 0; }