cf/Codeforces Round #373 div1-C/div2-E Sasha and Array 线段树 + 维护矩阵快速幂

http://codeforces.com/contest/718/problem/C

题意：

 n个数，维护两个操作，操作1，【l，r】区间每个位置+x，操作2，【l，r】区间内，以每个位置上的数字为下标的斐波那契数列的和,%1e9+7

 

那么线段树的节点都换成一个2X2的矩阵

 

sum【i】节点维护的是i所管辖的区间的答案（也就是A[l...r]里每一个Fib（A[i]）的矩阵的和）

add[i]表示节点对应区间 的lazy标记，代表整个区间加X，实际意义就相当于给区间里的每一个数fib(A【i】)再乘上一个 【系数矩阵的X次方矩阵】

update：

f为fib系数矩阵

由于 sum[i]类似于 A+B+C+D 

而我们要算的是A*(f^x)+B*( f^x)+ C*(f^x) +D* (f^x)

等价于 (A+B+C+D) *（f^x）

因此我们update时，直接对sum[i] = sum[i]* (f^x)即可

同理        add[i] = mul(add[i],val); 【系数矩阵的累乘，这个很显然了】

 

【1 1】

【1 0】作为矩阵基，那么sum[n].mat[0][0]实际代表fib(n+1),因此查询时返回sum[i].mat[0][1]即可.

 

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
const int Njz = 2;
const long long  mod=1e9+7;
struct Matrix
{
    int a[2][2];
    Matrix()
    {
        memset(a,0,sizeof(a));
    }
    void init()
    {
        for(int i=0; i<Njz; i++)
            for(int j=0; j<Njz; j++)
                a[i][j]=(i==j);
    }
    Matrix operator + (const Matrix &B)const
    {
        Matrix C;
        for(int i=0; i<Njz; i++)
            for(int j=0; j<Njz; j++)
                C.a[i][j]=(a[i][j]+B.a[i][j])%mod;
        return C;
    }
    Matrix operator * (const Matrix &B)const
    {
        Matrix C;
        for(int i=0; i<Njz; i++)
            for(int k=0; k<Njz; k++)
                for(int j=0; j<Njz; j++)
                    C.a[i][j]=(C.a[i][j]+1LL*a[i][k]*B.a[k][j])%mod;
        return C;
    }
    Matrix operator ^ (const int &t)const
    {
        Matrix A=(*this),res;
        res.init();
        int p=t;
        while(p)
        {
            if(p&1)res=res*A;
            A=A*A;
            p>>=1;
        }
        return res;
    }
} base,xishu;
 
const int N = 100005  ;
Matrix   add[4*N],sum[4*N];
void pushup(int rt)
{
    sum[rt]=  sum[rt<<1]+ sum[rt<<1|1] ;
}
void build(int l,int r,int i)    //  线段树的建立；
{
    add[i].init();
    if(l==r)
    {
        sum[i]=base;
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,i<<1);
    build(mid+1,r,i<<1|1);
    pushup(i);
}
void pushDown(int i, int l, int r)		//把i节点的延迟标记传递到左右儿子节点
{
    if(add[i].a[0][0] != 0)
    {
        int mid = (l + r) >> 1;
        add[i << 1] = add[i << 1]* add[i] ;
        sum[i << 1]  = (sum[i << 1]*  add[i]);   //[l, mid]代表左儿子区间
        add[i << 1 | 1] =(add[i << 1 | 1]* add[i]);
        sum[i << 1 | 1]  = (sum[i << 1 | 1]*add[i]);
        add[i] .init();
    }
}
void update(int i, int l, int r, int ql, int qr, Matrix& val) //更新区间为qlqr,当前区间为l,r,代表当前区间和的节点为i，更新值为val,
{
    if(l > qr || ql > r)		//更新区间不在当前区间内
        return ;
    if(l >= ql && r <= qr)	//要更新的区间把当前区间完全包括，则把当前整个区间+val，然后返回上一层
    {
        add[i] = (add[i]*val);
        sum[i]  = (sum[i]*val);
        return ;
    }
    pushDown(i, l, r);			//如果上面没reutrn 表示要往左右儿子区间查询，所以把延迟标记放下去
    int mid = (l + r) >> 1;
    update(i << 1, l, mid, ql, qr, val);
    update(i << 1 | 1, mid + 1, r, ql, qr, val);
    pushup(i);
}
 
long long query(int i, int l, int r, int ql, int qr)	 //查询区间为qlqr,当前区间为l,r,代表当前区间和的节点为i
{
    if(l > qr || ql > r)
        return 0;
    if(l >= ql && r <= qr)
        return sum[i].a[0][1];
    pushDown(i, l, r);				//同update
    int mid =( l + r) >> 1;
    return (query(i << 1, l, mid, ql, qr)
            + query(i << 1 | 1, mid + 1, r, ql, qr))%mod;
}
void pre_fib()
{
    base.a[0][0]=1,base.a[0][1]=1;
    base.a[1][0]=0,base.a[1][1]=0;
    xishu.a[0][0]=1,xishu.a[0][1]=1;
    xishu.a[1][0]=1,xishu.a[1][1]=0;
 
}
int aa[N];
int main()
{
    pre_fib();
 
    int n,a,b,m;
    scanf("%d%d", &n,&m);
    build(1,n,1);
    for(int  i = 1; i <= n; i++)
        scanf("%d",&aa[i]) ;
    Matrix tmp;
    for(int  i = 1; i <= n; i++)
    {
        tmp=xishu^(aa[i]-1);
        update(1, 1, n, i, i,tmp);
    }
    int op,c;
    for(  int i = 1; i <= m; i++)
    {
        scanf("%d%d%d",&op,&a,&b);
        if (op==1)
        {
            scanf("%d",&c);
            tmp=xishu^(c);
            update(1, 1, n, a, b,tmp);
        }
        else
            printf("%lld\n",query(1, 1,n,a,b));
    }
 
    printf("\n");
 
    return 0;
}