ZOJ 4009 And Another Data Structure Problem （线段树+循环节+标记永久化）

ZOJ 4009 And Another Data Structure Problem （线段树+循环节+标记永久化）

题意：

1个序列有n个元素，有2种操作

1. l,r 把[l，r]元素变成$a_l^3,a_{l+1}^3,a_{l+2}^3....a_r^3$

2. l,r 求[l，r]区间和，取模99971

思路：

显而易见， 任何数模99971有一个48位的循环节~~(不然这题没法做了)~~

然后建48棵线段树，维护每一个树在每个循环节的情况。用标记永久化，更新区间时，先更新目的区间的lazy，再更新沿路sum；询问时，沿路求lazy和，返回对应的sum

用标记永久化好处：代码更好写，不容易懂，想不出来

#include<bits/stdc++.h>
using namespace std;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
typedef long long ll;
const int N=1e5+5,maxn=4e6+5;
const ll mod=99971;
int t,n,q;
ll a[maxn],sum[maxn][50],lazy[maxn];

ll sqr(ll x){
	return (x*x%mod)*x%mod;
}
void build(int rt,int l,int r){
	lazy[rt]=0;
	if(l==r){
		sum[rt][0]=a[l]%mod;
		for(int i=1;i<48;i++)sum[rt][i]=sqr(sum[rt][i-1]);
		return;
	}
	int mid=(l+r)>>1;
	build(lson);
	build(rson);
	for(int i=0;i<48;i++)sum[rt][i]=(sum[rt<<1][i]+sum[rt<<1|1][i])%mod;
}
void update(int rt,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		lazy[rt]=(lazy[rt]+1)%48;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)update(lson,x,y);
	if(y>mid) update(rson,x,y);
	for(int i=0;i<48;i++)sum[rt][i]=(sum[rt<<1][(i+lazy[rt<<1])%48]+sum[rt<<1|1][(i+lazy[rt<<1|1])%48])%mod;
}
ll query(int rt,int l,int r,int x,int y,int he){
	he=(he+lazy[rt])%48;
	if(x<=l&&r<=y)return sum[rt][he];
	int mid=(l+r)>>1;
	ll res=0;
	if(x<=mid)res=(res+query(lson,x,y,he))%mod;
	if(y>mid)res=(res+query(rson,x,y,he))%mod;
	return res%mod;
}
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&q);
	    for(int i=1;i<=n;i++){
		    scanf("%d",&a[i]);
	    }
	    build(1,1,n);
	    while(q--){
	    	int op,x,y;
	    	scanf("%d%d%d",&op,&x,&y);
	    	if(op==1)update(1,1,n,x,y);
			else printf("%lld\n",query(1,1,n,x,y,0));
		}
	}
}