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