2018.10.09 ZYH的斐波那契数列(线段树+矩阵快速幂)

描述

ZYH最近研究数列研究得入迷啦!
现在有一个斐波拉契数列(f[1]=f[2]=1,对于n>2有f[n]=f[n-1]+f[n-2]),
但是斐波拉契数列太简单啦,于是ZYH把它改成了斐波拉契的前缀和的数列{Si}(S[1]=1,对于n>1,有S[n]=S[n-1]+f[n]),接下来ZYH要在{Si}数列上面做一些操作。
1.修改: 将数列中下标在[l,r]区间内的数加上v
2.查询:询问此时数列第k项对1000000007取模后是多少。
你能帮他解决这两个操作吗?

输入

第一行,一个正整数,n,代表操作的次数。
接下来n行,先输入一个数p,若p=1,则再输入三个正整数l,r,v,代表将数列中下标在[l,r]区间内的数加上v;若p=2,则再输入一个正整数k, 询问此时数列第k项mod1000000007后是多少。

输出

对于每次询问,输出对应的整数,每个询问输出一行。

样例输入

5
1 1 10 2
2 1
2 2
2 3
2 4

样例输出

3
4
6
9

提示

k<=10^9,n<=50000


每次修改显然不能直接搞。
注意到一个点在修改过后的值等于原值+差分数组前缀和
我们应该用线段树动态开点来维护差分数组前缀和。
然后用矩阵快速幂来维护原值。
代码:

#include<bits/stdc++.h>
#define int long long
#define N 1000005
#define M 1000000001
#define mod 1000000007
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int n,tot=1;
struct Node{int l,r;int val;Node(){l=r=0;val=0ll;}}T[N<<2];
struct Matrix{int a[3][3];};
inline Matrix operator*(Matrix a,Matrix b){
	Matrix c;
	memset(c.a,0,sizeof(c.a));
	for(int i=1;i<=2;++i)for(int k=1;k<=2;++k)for(int j=1;j<=2;++j)c.a[i][j]+=a.a[i][k]*b.a[k][j]%mod;
	for(int i=1;i<=2;++i)for(int j=1;j<=2;++j)c.a[i][j]%=mod;
	return c;
}
inline Matrix operator^(Matrix a,int p){
	Matrix ret;
	ret.a[1][1]=ret.a[2][2]=1,ret.a[1][2]=ret.a[2][1]=0;
	while(p){
		if(p&1)ret=ret*a;
		a=a*a,p>>=1;
	}
	return ret;
}
inline void pushup(int p){(T[p].val=T[T[p].l].val+T[T[p].r].val)%=mod;}
inline void update(int p,int l,int r,int k,int v){
	if(l==r){(T[p].val+=v)%=mod;return;}
	int mid=l+r>>1;
	if(k<=mid){
		if(!T[p].l)T[p].l=++tot;
		update(T[p].l,l,mid,k,v);
	}
	else{
		if(!T[p].r)T[p].r=++tot;
		update(T[p].r,mid+1,r,k,v);
	}
	pushup(p);
}
inline int query(int p,int l,int r,int ql,int qr){
	if(!p)return 0;
	if(ql<=l&&r<=qr)return T[p].val;
	int mid=l+r>>1;
	if(qr<=mid)return query(T[p].l,l,mid,ql,qr);
	if(ql>mid)return query(T[p].r,mid+1,r,ql,qr);
	return (query(T[p].l,l,mid,ql,mid)+query(T[p].r,mid+1,r,mid+1,qr))%mod;
}
signed main(){
	n=read();
	Matrix tmp1,tmp2;
	tmp1.a[1][1]=1,tmp1.a[1][2]=1,tmp1.a[2][1]=1,tmp1.a[2][2]=0;
	tmp2.a[1][1]=1,tmp2.a[1][2]=0,tmp2.a[2][1]=1,tmp2.a[2][2]=0;
	while(n--){
		int op=read();
		if(op==1){
			int l=read(),r=read();
			int v=read();
			update(1,1,M,l,v),update(1,1,M,r+1,-v);
		}
		else{
			int k=read();
			int ans=query(1,1,M,1,k);
			Matrix tmp=(tmp1^k)*tmp2;
			ans=(ans+tmp.a[1][1]-1+mod)%mod;
			cout<<ans<<'\n';
		}
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值