[CF718C]Sasha and Array

Sasha and Array

题解

水题线段树

又是一道序列操作，我们延续上一道的思想，继续用线段树。

我们还是用一个点来对应那一段区间的斐波拉契值的和，因为斐波拉契只与加法相关，我们可以对一段和单独进行斐波拉契的操作来求出那一段的斐波拉契和。

于是可以通过懒标记的方法来对其进行记录。

问题是如果在每次pushdown的时候再对其计算斐波拉契矩阵并乘上来时明显会T飞，所以我们每次记录懒标记时用一个矩阵来记录，每次进行操作时也需要用一个矩阵来乘。笔者因为这个T了很久。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long LL;
#define int LL
const int mo=1e9+7;
#define gc() getchar()
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=gc();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=gc();}
	while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=gc();}
	x*=f;
}
#define For(i,a,b) for(int i=(a);i<=(b);++i)
inline int add(int a,int b){return a+b>mo?a+b-mo:a+b;}
struct ming{
    LL c[3][3];
    ming(){memset(c,0,sizeof(c));}
    ming operator * (const ming& a){
        ming r;
        For(i,1,2)            
			For(k,1,2)
	           	For(j,1,2)
	                r.c[i][j]=add(r.c[i][j],c[i][k]*a.c[k][j]%mo);
        return r;
    }
    ming operator + (const ming& a){
    	ming r;
    	For(i,1,2)
    		For(j,1,2)
    			r.c[i][j]=add(r.c[i][j],add(c[i][j],a.c[i][j]));
    	return r;
	}
	void clear(){
		For(i,1,2)For(j,1,2)c[i][j]=0;
	}
	void init(){
		for(int i=1;i<3;i++)c[i][i]=1; 
	} 
	bool empty(){
		if(c[1][1]!=1)return 0;
		if(c[1][2]!=0)return 0;
		if(c[2][1]!=0)return 0;
		if(c[2][2]!=1)return 0;
		return 1;
	}
	void print(){
        For(i,1,2){
            For(j,1,2){
                if(j!=1)putchar(' ');
                printf("%lld",c[i][j]);
            }
            putchar('\n');
        }
    }
}B[400000],lzy[400000];
int ad[100005];
ming qkpow(ming a,LL s){
	ming res;
	For(i,1,2) res.c[i][i]=1;
	while(s>0){
		if(s&1) res=res*a;
		a=a*a;s>>=1;
	}
	return res;
}
ming fibonacci(int b){
	ming C;
	C.c[1][2]=C.c[2][1]=C.c[2][2]=1;
	return qkpow(C,b);
}
void build(int rt,int l,int r){
	B[rt].clear();lzy[rt].clear();lzy[rt].init();
	if(l>r) return ;
	if(l==r){
		B[rt].c[1][1]=0;B[rt].c[1][2]=1;
		B[rt]=B[rt]*fibonacci(ad[l]);
		return ;
	}
	int mid=l+r>>1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	B[rt]=B[rt<<1]+B[rt<<1|1];
	//printf("build%d %d %d\n",rt,l,r);
	//B[rt].print();puts("\n");
}
void pushdown(int rt){
	if(!lzy[rt].empty()){
		B[rt<<1]=B[rt<<1]*lzy[rt];lzy[rt<<1]=lzy[rt<<1]*lzy[rt];
		B[rt<<1|1]=B[rt<<1|1]*lzy[rt];lzy[rt<<1|1]=lzy[rt<<1|1]*lzy[rt];
		lzy[rt].clear();lzy[rt].init();
	}
}
void insert(int rt,int l,int r,int al,int ar,ming aw){
	if(l>ar||r<al||l>r) return ;
	if(al<=l&&r<=ar){
		B[rt]=B[rt]*aw;lzy[rt]=lzy[rt]*aw;
		return ;
	}
	int mid=(l+r)>>1;pushdown(rt);
	insert(rt<<1,l,mid,al,ar,aw);
	insert(rt<<1|1,mid+1,r,al,ar,aw);
	B[rt]=B[rt<<1]+B[rt<<1|1];
	//printf("insert%d %d %d\n",rt,l,r);
	//B[rt].print();puts("\n");
}
ming query(int rt,int l,int r,int al,int ar){
	ming res;res.clear();
	if(l>ar||r<al||l>r)return res;
	if(al<=l&&r<=ar) return B[rt];
	int mid=(l+r)>>1;pushdown(rt);
	return query(rt<<1,l,mid,al,ar)+query(rt<<1|1,mid+1,r,al,ar);
}
int n,m;
signed main(){ 
	read(n);read(m);
	For(i,1,n)read(ad[i]);
	build(1,1,n);
	For(i,1,m){
		int opt;read(opt);
		if(opt==1){
			int l,r,x;read(l);read(r);read(x);
			ming w=fibonacci(x);insert(1,1,n,l,r,w);
		}
		else{
			int l,r;read(l);read(r);
			ming C=query(1,1,n,l,r);
			printf("%d\n",C.c[1][1]);
		}
	}
    return 0;
}

谢谢！！！