Codeforces 446C 线段树 递推Fibonacci公式

#include <iostream>
#include <cstdio>
#include <cstring>
#define LL __int64
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
const LL M=1000000009;
const int N=300000+20;
LL F[N],K[N],G[N],A[N];
int n,m;
LL d[N<<2],f[N<<2],f1[N<<2],f2[N<<2];
void init()
{
    F[0]=G[0]=0;
    K[0]=1;
    F[1]=1;
    G[1]=0;
    K[1]=0;
    for (int i=2;i<=n+10;i++){
        F[i]=F[i-1]+F[i-2];
        K[i]=K[i-1]+K[i-2];
        G[i]=G[i-1]+F[i-1];
        if (F[i]>=M) F[i]%=M;
        if (K[i]>=M) K[i]%=M;
        if (G[i]>=M) G[i]%=M;
    }
    for (int i=1;i<=n;i++) scanf("%d",&A[i]);
}
void up(int rt)
{
    d[rt]=d[rt<<1]+d[rt<<1|1];
    if (d[rt]>=M) d[rt]%=M;
}
void build(int rt,int l,int r)
{
    f[rt]=0;
    if (l>=r){
        d[rt]=A[l]%M;
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    up(rt);
}
void fix(LL v1,LL v2,int L,int R,int rt,int l,int r);
void pushdown(int rt,int l,int r)
{
    if (l>=r) return;
    if (!f[rt]) return;
    int mid=(l+r)>>1;
    fix(f1[rt],f2[rt],l,mid,lson);

    LL t1=K[mid-l+1]*f1[rt]%M+K[mid-l+2]*f2[rt]%M;
    LL t2=K[mid-l+2]*f1[rt]%M+K[mid-l+3]*f2[rt]%M;
    if (t1>=M) t1%=M;
    if (t2>=M) t2%=M;
    fix(t1,t2,mid+1,r,rson);
    f[rt]=0;

}
void fix(LL v1,LL v2,int L,int R,int rt,int l,int r)
{
    if (L==l && r==R){ //采用这种锁定区间的方法是因为下面有个地方要用L计算，我之前的那种写法会出错
        d[rt]+=(v1*F[R-L+1]%M+G[R-L+1]*v2%M)%M;
        if (d[rt]>=M) d[rt]%=M;
        if (!f[rt]){
            f1[rt]=v1%M;
            f2[rt]=v2%M;
            f[rt]=1;
        }
        else{
            f1[rt]+=v1%M;
            f2[rt]+=v2%M;
            if (f1[rt]>=M) f1[rt]%=M;
            if (f2[rt]>=M) f2[rt]%=M;
        }
        return;
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    if (R<=mid) fix(v1,v2,L,R,lson);
    else if (L>mid) fix(v1,v2,L,R,rson);
    else {
        fix(v1,v2,L,mid,lson);
        LL t1=K[mid-L+1]*v1%M+K[mid-L+2]*v2%M;//这里L要进行计算，每个孩子对应唯一L，所以为什么我要采用这种锁定区间的方式，就是因为这里。以前以为两种锁定区间的方法差不多，现在找到区别了，如果只是简单为了锁定区间，两种都可以用，但是当区间要作为计算条件的时候，要采取这种方法避免错误
        LL t2=K[mid-L+2]*v1%M+K[mid-L+3]*v2%M;
        if (t1>=M) t1%=M;
        if (t2>=M) t2%=M;
        fix(t1,t2,mid+1,R,rson);
    }
    up(rt);
}
LL query(int L,int R,int rt,int l,int r)
{
    if (L<=l && r<=R){
        return d[rt]%M;
    }
    pushdown(rt,l,r);
    int mid=(l+r)>>1;
    LL ret1=0,ret2=0;
    if (L<=mid) ret1=query(L,R,lson);
    if (R>mid) ret2=query(L,R,rson);
    return (ret1+ret2)%M;
}
int main()
{
    int op,a,b;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        build(1,1,n);
        while (m--)
        {
            scanf("%d%d%d",&op,&a,&b);
            if (op==1){
                fix(1,1,a,b,1,1,n);
            }
            else{
                LL ans=query(a,b,1,1,n);
                printf("%I64d\n",ans);
            }
        }
    }
    return 0;
}