codeforces718C Sasha and Array
Description
维护一个序列,支持区间加,与求 ∑ri=lf(a) ,其中 f 为斐波那契。
题解
线段树真的是只要能合并,打标记什么都可维护呀!这道题目维护的就是fib的转移矩阵,就连打的标记都是矩阵。为什么可以合并?
因为矩阵满足
然后,然后就没有然后了。用矩阵维护这个序列就可以了。
#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;
}