hdu 6155 线段树加矩阵快速幂

http://acm.hdu.edu.cn/showproblem.php?pid=6155

第一次做矩阵快速幂套线段树的题，感觉好神奇

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5,mo=1e9+7;
struct NODE
{
long long a[3][3];
}t1,t2,be;
NODE mult(NODE a,NODE b)
{
NODE c;
memset(c.a,0,sizeof(c.a));
for(int i=0;i<3;i++)
{
for(int j=0;j<3;j++)
{
for(int k=0;k<3;k++)
{
c.a[i][j]+=a.a[i][k]*b.a[k][j];
if(c.a[i][j]>=mo)
c.a[i][j]%=mo;
}
}
}
return c;
}
NODE seg[4*maxn];
bool lazy[4*maxn];
int num[maxn];
void pushup(int node)
{
seg[node]=mult(seg[node<<1],seg[node<<1|1]);
}
void change(NODE &a)
{
for(int i=0;i<3;i++)
swap(a.a[i][0],a.a[i][1]);
for(int i=0;i<3;i++)
swap(a.a[0][i],a.a[1][i]);
}
void pushdown(int node)
{
if(lazy[node])
{
lazy[node<<1]=!lazy[node<<1];
lazy[node<<1|1]=!lazy[node<<1|1];
change(seg[node<<1]);
change(seg[node<<1|1]);
lazy[node]=0;
}
}
void build(int l,int r,int node)
{
lazy[node]=0;
if(l==r)
{
seg[node]=num[l]?t2:t1;
return ;
}
int m=(l+r)>>1;
build(l,m,node<<1);
build(m+1,r,node<<1|1);
pushup(node);
return ;
}
void update(int le,int ri,int l,int r,int node)
{
if(l>=le&&r<=ri)
{
lazy[node]=!lazy[node];
change(seg[node]);
return ;
}
pushdown(node);
int m=(l+r)>>1;
if(m>=le)
update(le,ri,l,m,node<<1);
if(m<ri)
update(le,ri,m+1,r,node<<1|1);
pushup(node);
return ;
}
NODE query(int le,int ri,int l,int r,int node)
{
if(l>=le&&r<=ri)
{
return seg[node];
}
pushdown(node);
NODE sum;
memset(sum.a,0,sizeof sum.a);
for(int i=0;i<3;i++)
sum.a[i][i]=1;
int m=(l+r)>>1;
if(m>=le)
sum=mult(sum,query(le,ri,l,m,node<<1));
if(m<ri)
sum=mult(sum,query(le,ri,m+1,r,node<<1|1));
return sum;
}
int ch[maxn];
int main()
{
t1.a[0][0]=t1.a[1][0]=t1.a[2][0]=t1.a[1][1]=t1.a[2][2]=1;
t2.a[0][0]=t2.a[0][1]=t2.a[1][1]=t2.a[2][1]=t2.a[2][2]=1;
be.a[0][2]=1;
int T,n,q,ty,l,r;
NODE ans;
cin>>T;
while(T--)
{
cin>>n>>q;
for(int i=1;i<=n;i++)
scanf("%1d",&num[i]);
build(1,n,1);
while(q--)
{
scanf("%d %d %d",&ty,&l,&r);
if(ty-1)
{
ans=mult(be,query(l,r,1,n,1));
printf("%lld\n",(ans.a[0][0]+ans.a[0][1])%mo);
}
else
{
update(l,r,1,n,1);
}
}
}
return 0;
}