CF 316E3 Summer Homework(斐波那契矩阵+线段树)

题目链接：http://codeforces.com/problemset/problem/316/E3

题意：一个数列A三种操作：（1）1 x y将x位置的数字修改为y；（2）2 x y求[x,y]区间的数字的和，和函数为如下；（3）3 x y z将[x,y]区间的数字统一加z。

思路：很明显，这是一个线段树的题目。其中第一种和第三种操作都很容易搞定，第一种直接更新到底；第三种增加标记。显然为了求和，我们必须要在节点上增加一个和，表示以这个节点为根的子树的数字的和函数，这里我们用a来表示。那么，到这里有两件事情要解决：

（1）如何在区间增加一个值x时更新a？设该区间为[L,R]，则增加的和为：

（2）已知两段数字的和函数，如何拼接成一个和函数？设前一段为3，后一段为4，那么我们要将后一段修改，修改前为：

修改后为：

为了方便，我们先看看怎么把（1,1,2,3）变为（1,2,3,5），也就是：

其实就是：

我们用a表示某段内（设这一段有x个数）x个数的和函数，b表示后x-1个数的和函数，那么上面其实就是a+b，得到A，那么B怎么得到呢？（AB和ab对应，我们发现B=a），因此我们得到转移矩阵：

注意这个的b和B是为了方便计算我们要新加到线段树节点中的值。那么上面(1,1,2,3)到(3,5,8,13)的转变岂不是成了：

到这里，我们就解决了两段和函数拼成一段的方法。

struct Matrix
{
    i64 a[2][2];

void init(int x)
    {
        clr(a,0);
        if(x==1) a[0][0]=a[1][1]=1;
        else if(x==2) a[0][0]=a[0][1]=a[1][0]=1;
    }

Matrix operator*(Matrix p)
    {
        Matrix ans;
        ans.init(0);
        int i,j,k;
        FOR0(k,2) FOR0(i,2) FOR0(j,2)
        {
            ans.a[i][j]+=a[i][k]*p.a[k][j];
            ans.a[i][j]%=mod;
        }
        return ans;
    }
};

Matrix A[N];
i64 f[N];

void init()
{
    A[0].init(1);
    A[1].init(2);
    int i;
    for(i=2;i<N;i++) A[i]=A[i-1]*A[1];

f[0]=f[1]=1;
    for(i=2;i<N;i++) f[i]=(f[i-1]+f[i-2])%mod;
    for(i=1;i<N;i++) f[i]=(f[i]+f[i-1])%mod;
}

struct node
{
    int L,R;
    i64 a,b,flag;

node(){}
    node(i64 _a,i64 _b)
    {
        a=_a;
        b=_b;
    }

node det(int t)
    {
        i64 x=(a*A[t].a[0][0]+b*A[t].a[0][1])%mod;
        i64 y=(a*A[t].a[1][0]+b*A[t].a[1][1])%mod;
        return node(x,y);
    }

void add(i64 x)
    {
        flag=(flag+x)%mod;
        a=(a+f[R-L]*x)%mod;
        if(R>L) b=(b+f[R-L-1]*x)%mod;
    }

node operator+(node p)
    {
        i64 x=(a+p.a)%mod,y=(b+p.b)%mod;
        return node(x,y);
    }
};

node a[N<<2];
int b[N];

void pushUp(int t)
{
    if(a[t].L==a[t].R) return;
    int mid=(a[t].L+a[t].R)>>1;
    node temp=a[t*2+1].det(mid-a[t].L+1);
    a[t].a=(a[t*2].a+temp.a)%mod;
    a[t].b=(a[t*2].b+temp.b)%mod;
}

void pushDown(int t)
{
    if(a[t].L==a[t].R) return;
    if(a[t].flag)
    {
        a[t*2].add(a[t].flag);
        a[t*2+1].add(a[t].flag);
        a[t].flag=0;
    }
}

void build(int t,int L,int R)
{
    a[t].L=L;
    a[t].R=R;
    a[t].flag=a[t].b=0;
    if(L==R)
    {
        a[t].a=b[L];
        return;
    }
    int mid=(a[t].L+a[t].R)>>1;
    build(t*2,L,mid);
    build(t*2+1,mid+1,R);
    pushUp(t);
}

void change(int t,int pos,int x)
{
    if(a[t].L==a[t].R)
    {
        a[t].a=x;
        a[t].b=0;
        a[t].flag=0;
        return;
    }
    pushDown(t);
    int mid=(a[t].L+a[t].R)>>1;
    if(pos<=mid) change(t*2,pos,x);
    else change(t*2+1,pos,x);
    pushUp(t);
}

void change(int t,int L,int R,int x)
{
    if(a[t].L==L&&a[t].R==R)
    {
        a[t].add(x);
        return;
    }
    pushDown(t);
    int mid=(a[t].L+a[t].R)>>1;
    if(R<=mid) change(t*2,L,R,x);
    else if(L>mid) change(t*2+1,L,R,x);
    else
    {
        change(t*2,L,mid,x);
        change(t*2+1,mid+1,R,x);
    }
    pushUp(t);
}

node query(int t,int L,int R)
{
    if(a[t].L==L&&a[t].R==R) return a[t];
    pushDown(t);
    node A,B;
    int mid=(a[t].L+a[t].R)>>1;
    if(R<=mid) A=query(t*2,L,R);
    else if(L>mid) A=query(t*2+1,L,R);
    else
    {
        A=query(t*2,L,mid);
        B=query(t*2+1,mid+1,R);
        A=A+B.det(mid-L+1);
    }
    pushDown(t);
    return A;
}

int n,m;

int main()
{
    init();
    Rush(n)
    {
        RD(m);
        int i;
        FOR1(i,n) RD(b[i]);
        build(1,1,n);
        int op,x,y,z;
        while(m--)
        {
            RD(op);
            if(op==1)
            {
                RD(x,y);
                change(1,x,y);
            }
            else if(op==2)
            {
                RD(x,y);
                PR(query(1,x,y).a);
            }
            else
            {
                RD(x,y,z);
                change(1,x,y,z);
            }
        }
    }
}