codeforces718C Sasha and Array

codeforces718C Sasha and Array

Description

维护一个序列，支持区间加，与求$\sum_{i=l}^{r}f(a)$，其中$f$为斐波那契。

题解

线段树真的是只要能合并，打标记什么都可维护呀！这道题目维护的就是fib的转移矩阵，就连打的标记都是矩阵。为什么可以合并？
因为矩阵满足 $A*C+B*C=(A+B)*C$
然后，然后就没有然后了。用矩阵维护这个序列就可以了。

#include <cstdio>
#include <algorithm>
#include <cstring>
#define MAXN 800010
#define MOD 1000000007
#define LL long long
using namespace std;
int n,m;
LL a[MAXN];
struct M{
    LL date[3][3];
    void init() {date[1][1]=1;date[2][1]=0;date[1][2]=0;date[2][2]=1;}
    M operator + (const M &a) const 
    {
        M tmp;
        tmp.date[1][1]=(date[1][1]+a.date[1][1])%MOD;
        tmp.date[1][2]=(date[1][2]+a.date[1][2])%MOD;
        tmp.date[2][1]=(date[2][1]+a.date[2][1])%MOD;
        tmp.date[2][2]=(date[2][2]+a.date[2][2])%MOD;   
        return tmp;
    }
    M operator * (const M &a) const
    {   
        M tmp;
        tmp.date[1][1]=(date[1][1]*a.date[1][1]+date[1][2]*a.date[2][1])%MOD;
        tmp.date[1][2]=(date[1][1]*a.date[2][1]+date[1][2]*a.date[2][2])%MOD;
        tmp.date[2][1]=(date[2][1]*a.date[1][1]+date[2][2]*a.date[2][1])%MOD;
        tmp.date[2][2]=(date[2][1]*a.date[1][2]+date[2][2]*a.date[2][2])%MOD;
        return tmp;
    }
}sum[MAXN],fib,add[MAXN];
M power(M a,LL k)
{
    M ans;ans.init();
    while(k)
    {
        if(k&1) ans=ans*a;
        a=a*a;
        k>>=1;
    }
    return ans;
}
void pushup(int rt)
{   
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt)
{   
    if(!(add[rt].date[1][1]==1&&add[rt].date[2][2]==1
    &&add[rt].date[1][2]==0&&add[rt].date[2][1]==0))
    {
        sum[rt<<1]=sum[rt<<1]*add[rt];
        sum[rt<<1|1]=sum[rt<<1|1]*add[rt];
        add[rt<<1]=add[rt<<1]*add[rt];
        add[rt<<1|1]=add[rt<<1|1]*add[rt];
        add[rt].init();
    }
}
void build(int l,int r,int rt)
{
    add[rt].init();
    if(l==r)
    {   
        sum[rt]=power(fib,a[l]);
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(int L,int R,int x,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        M tmp=power(fib,x);
        sum[rt]=sum[rt]*tmp;
        add[rt]=add[rt]*tmp;
        return ;
    }
    int mid=(l+r)>>1;
    pushdown(rt);
    if(L<=mid) update(L,R,x,l,mid,rt<<1);
    if(mid<R) update(L,R,x,mid+1,r,rt<<1|1);
    pushup(rt);
}
M query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)  return sum[rt];
    int mid=(l+r)>>1;M ans;
    ans.date[1][1]=ans.date[1][2]=ans.date[2][1]=ans.date[2][2]=0;
    pushdown(rt);
    if(L<=mid) ans=ans+query(L,R,l,mid,rt<<1);
    if(mid<R) ans=ans+query(L,R,mid+1,r,rt<<1|1);
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    fib.date[1][1]=fib.date[1][2]=fib.date[2][1]=1;fib.date[2][2]=0;
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    build(1,n,1);
    int opt,x,y,z;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&opt,&x,&y);
        if(opt==1)
        {
            scanf("%d",&z);
            update(x,y,z,1,n,1);
        }
        else
        if(opt==2)
        {   
            M ans;
            ans=query(x,y,1,n,1);
            printf("%lld\n",ans.date[1][2]);
        }
    }
    return 0;
}