CF 580E Kefa and Watch（线段树维护hash）

题目链接：https://www.luogu.com.cn/problem/CF580E

#include<bits/stdc++.h>
#define ull unsigned long long
using namespace std;
const int N=1000011;
const ull St=233,M1=1000000007,M2=1000000009;
int n,m,k;
ull jc[N][2],jcs[N][2];
char s[N];

struct Node {
    int l,r;
    ull data[2],tag[2];
    bool vis;
};
Node t[N];

inline int re_ad() {
    char ch=getchar(); int x=0,f=1;
    while(ch<'0' || ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while('0'<=ch && ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;
}

inline void init() {
    jc[0][0]=1,jc[0][1]=1,jcs[0][0]=1,jcs[0][1]=1;
    for(int i=1;i<=n;++i) (jc[i][0]=jc[i-1][0]*St)%=M1,(jc[i][1]=jc[i-1][1]*St)%=M2;
    for(int i=1;i<=n;++i) (jcs[i][0]=jcs[i-1][0]+jc[i][0])%=M1,(jcs[i][1]=jcs[i-1][1]+jc[i][1])%=M2;
}

inline void pushup(int d) {
    int mid=(t[d].l+t[d].r)>>1;
    (t[d].data[0]=t[d<<1].data[0]*jc[t[d].r-mid][0]+t[d<<1|1].data[0])%=M1;
    (t[d].data[1]=t[d<<1].data[1]*jc[t[d].r-mid][1]+t[d<<1|1].data[1])%=M2;
}

inline void pushdown(int d) {
    if(t[d].vis==false) return ;
    (t[d<<1].tag[0]=t[d].tag[0])%=M1;
    (t[d<<1].tag[1]=t[d].tag[1])%=M2;
    t[d<<1].vis=true;
    (t[d<<1|1].tag[0]=t[d].tag[0])%=M1;
    (t[d<<1|1].tag[1]=t[d].tag[1])%=M2;
    t[d<<1|1].vis=true;
    (t[d<<1].data[0]=t[d].tag[0]*jcs[t[d<<1].r-t[d<<1].l][0])%=M1;
    (t[d<<1].data[1]=t[d].tag[1]*jcs[t[d<<1].r-t[d<<1].l][1])%=M2;
    (t[d<<1|1].data[0]=t[d].tag[0]*jcs[t[d<<1|1].r-t[d<<1|1].l][0])%=M1;
    (t[d<<1|1].data[1]=t[d].tag[1]*jcs[t[d<<1|1].r-t[d<<1|1].l][1])%=M2;
    t[d].tag[0]=0,t[d].tag[1]=0,t[d].vis=false;
}

void build(int d,int x,int y) {
    t[d].l=x,t[d].r=y,t[d].vis=false;
    if(x==y) {
        (t[d].data[0]=s[x-1]-'0')%=M1;
        (t[d].data[1]=s[x-1]-'0')%=M2;
        return ;
    }
    int mid=(x+y)>>1;
    build(d<<1,x,mid);
    build(d<<1|1,mid+1,y);
    pushup(d);
}

void update(int d,int x,int y,ull z) {
    if(t[d].r<x || t[d].l>y) return ;
    if(x<=t[d].l && t[d].r<=y) {
        t[d].vis=true;
        (t[d].tag[0]=z)%=M1;
        (t[d].tag[1]=z)%=M2;
        (t[d].data[0]=z*jcs[t[d].r-t[d].l][0])%=M1;
        (t[d].data[1]=z*jcs[t[d].r-t[d].l][1])%=M2;
        return ;
    }
    pushdown(d);
    int mid=(t[d].l+t[d].r)>>1;
    if(mid>=x) update(d<<1,x,y,z);
    if(mid+1<=y) update(d<<1|1,x,y,z);
    pushup(d);
}

ull query1(int d,int x,int y) {
    if(t[d].r<x || t[d].l>y) return (ull)0;
    if(x<=t[d].l && t[d].r<=y) {
        return t[d].data[0];
    }
    pushdown(d);
    int mid=(t[d].l+t[d].r)>>1;
    ull res=0;
    if(mid<x) return query1(d<<1|1,x,y);
    else if(mid>=y) return query1(d<<1,x,y);
    return (query1(d<<1,x,mid)*jc[y-mid][0]+query1(d<<1|1,mid+1,y))%M1;
}

ull query2(int d,int x,int y) {
    if(t[d].r<x || t[d].l>y) return (ull)0;
    if(x<=t[d].l && t[d].r<=y) {
        return t[d].data[1];
    }
    pushdown(d);
    int mid=(t[d].l+t[d].r)>>1;
    ull res=0;
    if(mid<x) return query2(d<<1|1,x,y);
    else if(mid>=y) return query2(d<<1,x,y);
    return (query2(d<<1,x,mid)*jc[y-mid][1]+query2(d<<1|1,mid+1,y))%M2;
}

inline void Query(int x,int y,int z) {
    if(z>y-x+1) {
        printf("NO\n");
        return ;
    }
    ull res10=query1(1,x,y-z),res11=query2(1,x,y-z);
//    printf("%d %d %llu %llu ",x,y-z,res10,res11);
    ull res20=query1(1,x+z,y),res21=query2(1,x+z,y);
//    printf("%d %d %llu %llu\n",x+z,y,res20,res21);
    (res10==res20 && res11==res21) ? printf("YES\n") : printf("NO\n");
}

int main()
{
    int goal,x,y,z;
    n=re_ad(),m=re_ad(),k=re_ad();
    init();
    scanf("%s",s);
    build(1,1,n);
    for(int i=1;i<=m+k;++i) {
        goal=re_ad(),x=re_ad(),y=re_ad(),z=re_ad();
        if(goal==1) update(1,x,y,(ull)z);
        else Query(x,y,z);
    }
//    for(int i=1;i<=n;++i) printf("%llu ",jc[i]); puts("");
//    for(int i=1;i<=n;++i) printf("%llu ",jcs[i]); puts("");
//    for(int i=1;i<=10;++i) printf("%d %d %llu\n",t[i].l,t[i].r,t[i].data);
    return 0; 
}