CodeForces 719E Sasha and Array(线段树+矩阵快速幂)

终于做到线段树维护矩阵的题目了，本题没有斐波那契数列的话就是一道简单的线段树板子题，由于更新的x可能很大，所以递推求fib(n)肯定会TLE，考虑矩阵快速幂来维护，最后只需要取矩阵x[0][0]的元素即可，注意细节，易错点见代码中注释。

#include<bits/stdc++.h>
#define lowbit(x) ((x)&(-(x)))
#define ll long long
#define INF 2147483647
#define CLR(a) memset(a, 0, sizeof(a))
using namespace std;
#define ls (rt<<1)
#define rs (rt<<1|1)
#define MID(a,b) (a+((b-a)>>1))
const int mod=1e9+7;
const int maxn=1e5+10;
int t[maxn];
struct Mat{
	ll x[2][2];
	void init(){//单位阵 
		x[0][0]=x[1][1]=1;
		x[1][0]=x[0][1]=0;
	}
	Mat operator*(const Mat& a)const{
		Mat ans;
		ans.x[0][0]=ans.x[0][1]=ans.x[1][0]=ans.x[1][1]=0;
		for(int k=0;k<2;k++)
			for(int i=0;i<2;i++)
				for(int j=0;j<2;j++)
					ans.x[i][j]=(ans.x[i][j]+x[i][k]*a.x[k][j])%mod;
		return ans;
	}
	Mat operator+(const Mat& a)const{
		Mat ans;
		for(int i=0;i<2;i++)
			for(int j=0;j<2;j++)
				ans.x[i][j]=(x[i][j]+a.x[i][j])%mod;
		return ans;
	}
};
Mat pow(ll n){
	Mat m,ret;
	m.x[0][0]=m.x[0][1]=m.x[1][0]=1;
	m.x[1][1]=0;
	ret.init();//WA 初始化
	while(n){
		if(n&1) ret=ret*m;
		m=m*m;
		n>>=1; //WA 少了等号，检查了半天
	}
	return ret;
}
struct node{
	int l,r; 
	int mid(){return MID(l,r);}
	Mat lazy,sum;
};
struct Segtree{
	node tree[maxn<<2];
	void push_up(int rt){
		tree[rt].sum=tree[ls].sum+tree[rs].sum;
	}
	void push_down(int rt){
		tree[ls].sum=tree[rt].lazy*tree[ls].sum;
		tree[rs].sum=tree[rt].lazy*tree[rs].sum;
		tree[ls].lazy=tree[rt].lazy*tree[ls].lazy;
		tree[rs].lazy=tree[rt].lazy*tree[rs].lazy;
		tree[rt].lazy.init();
	}
	void build(int rt,int l,int r){
		tree[rt].l=l;tree[rt].r=r; 
		tree[rt].lazy.init();
		tree[rt].sum.init();
		if(l==r){
			tree[rt].sum=pow(t[l]-1);//WA 一定要-1。
		}
		else{
			int mid=tree[rt].mid();
			build(ls,l,mid);
			build(rs,mid+1,r);
			push_up(rt);
		}
	}
	void update(int rt,int st,int ed,Mat val){
		int l=tree[rt].l,r=tree[rt].r;
		if(st<=l&&r<=ed){
			tree[rt].sum=tree[rt].sum*val; //WA漏写了
			tree[rt].lazy=tree[rt].lazy*val;
		}
		else{
			push_down(rt);
			int mid=tree[rt].mid();
			if(st<=mid) update(ls,st,ed,val);
			if(ed>mid) update(rs,st,ed,val);
			push_up(rt);
		}
	}
	ll query(int rt,int st,int ed){
		int l=tree[rt].l,r=tree[rt].r;
		if(st<=l&&r<=ed) return tree[rt].sum.x[0][0];
		else{
			push_down(rt);
			int mid=tree[rt].mid();
			ll ans=0;
			if(st<=mid) ans=(ans+query(ls,st,ed))%mod;
			if(ed>mid) ans=(ans+query(rs,st,ed))%mod;
			return ans;
		} 
	}
}seg;
int n,m;
int main() {
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&t[i]);
	seg.build(1,1,n);
	for(int i=1;i<=m;i++){
		int op,a,b,c;
		scanf("%d",&op);
		if(op==1){
			scanf("%d%d%d",&a,&b,&c);
			Mat res=pow(c);
			seg.update(1,a,b,res);
		}else{
			scanf("%d%d",&a,&b);
			ll fin=seg.query(1,a,b);
			printf("%lld\n",fin);
		}
	}
	return 0;
}