2020CCPC威海站 G Caesar Cipher （线段树维护哈希）

2020CCPC威海站 G Caesar Cipher （线段树维护哈希）

题意：

给定你n和q，然后给定你一个长度为n的序列，q次操作，每次操作有两种形式：

1 x y:代表从序列的第x项到第y项所有的(a[i] + 1) mod 65536

2 x y l:表示从询问分别以x和y为起点的两个序列，然后这两个序列是否完全相同，输出yes或no

思路：

对于操作1，可以直接用线段树维护

对于操作2，多加一个hash数组进去，存储每个区间的hash值。判断时，通过比较区间的hash值，从而判断是否相等。

对于取模，写一个函数用来判断是否需要取模，以及取模后的更新。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
const int mod=65536;
const int maxn=5e5+5;
const int P=13331;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
ll n,m,q,lazy[maxn<<2],maxx[maxn<<2],a[maxn],base[maxn<<2],hx[maxn<<2],f[maxn];
void init(){
	f[0]=1;
	for(int i=1;i<=maxn;i++)f[i]=f[i-1]*P%MOD;
}
void up(int rt){
	maxx[rt]=max(maxx[rt<<1],maxx[rt<<1|1]);
    hx[rt]=(hx[rt<<1]+hx[rt<<1|1])%MOD;
}
void pushdown(int rt,int l,int r){
	if(lazy[rt]){
		maxx[rt<<1]+=lazy[rt];
		maxx[rt<<1|1]+=lazy[rt];
		hx[rt<<1]=(hx[rt<<1]+lazy[rt]*base[rt<<1])%MOD;
		hx[rt<<1|1]=(hx[rt<<1|1]+lazy[rt]*base[rt<<1|1])%MOD;
		lazy[rt<<1]+=lazy[rt];
		lazy[rt<<1|1]+=lazy[rt];
		lazy[rt]=0;
	}
}
void build(int rt,int l,int r){
	if(l==r){
		maxx[rt]=a[l];
		lazy[rt]=0;
		base[rt]=f[l];
		hx[rt]=a[l]*f[l]%MOD;
		return;
	}
	int mid=(l+r)>>1;
	build(lson);
	build(rson);
	up(rt);
	base[rt]=(base[rt<<1]+base[rt<<1|1])%MOD;
}
void update(int rt,int l,int r,int x,int y){
	if(x<=l&&r<=y){
    	maxx[rt]++;
    	lazy[rt]++;
    	hx[rt]=(hx[rt]+base[rt])%MOD;
        return;
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    if(x<=mid)
        update(lson,x,y);
    if(y>mid)
	    update(rson,x,y);
    up(rt);
}
void upmod(int rt,int l,int r,int x,int y){
	if(maxx[rt]<mod)return;
	if(l==r){
		maxx[rt]%=mod,hx[rt]=(maxx[rt]*base[rt])%MOD;
	//	lazy[rt]=0;
		return; 
	}
	int mid=(l+r)>>1;
	pushdown(rt,l,r);
	if(x<=mid)upmod(lson,x,y);
	if(y>mid)upmod(rson,x,y);
	up(rt);
}
ll query(int rt,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		return hx[rt];
	}
	int mid=(l+r)>>1;
	ll ans=0;
	pushdown(rt,l,r);
	if(x<=mid)ans=(ans+query(lson,x,y))%MOD;
	if(y>mid)ans=(ans+query(rson,x,y))%MOD;
	return ans%MOD;
}
int main(){
	init();
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	}
	build(1,1,n);
	while(m--){
		int op,x,y,z;
		scanf("%d",&op);
		if(op==1){
			scanf("%d%d",&x,&y);
			update(1,1,n,x,y);
			upmod(1,1,n,x,y);
		}
		else{
			scanf("%d%d%d",&x,&y,&z);
			if(x>y)swap(x,y);
			ll t1=query(1,1,n,x,x+z-1),t2=query(1,1,n,y,y+z-1);
			if(t1*f[y-x]%MOD==t2){
				printf("yes\n");
			}
			else printf("no\n");
		}
	}
}