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;}