E. Kefa and Watch 题解(线段树维护hash+循环节结论)

题目链接

题目大意

对数组又2种操作

1、区间整体赋值

2、询问[l,r]是否有长度为d的循环节

题目思路

其实就是一个结论

如果s[l,r-d]==s[l+d,r],那么就有长度为d的循环节，自己画图演示即可发现

然后线段树维护hash即可

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define fi first
#define se second
#define debug printf("aaaaaaaaaaa\n");
const int maxn=1e5+5,inf=0x3f3f3f3f,mod=1e9+7;
const ll INF=0x3f3f3f3f3f3f3f3f;
int n,m,k;
char s[maxn];
ull tree[maxn<<2],lazy[maxn<<2];
ull pre[maxn],base=233,prepre[maxn];
void updown(int node,int l,int r){
    if(lazy[node]){
        int mid=(l+r)/2;
        lazy[node<<1]=lazy[node];
        lazy[node<<1|1]=lazy[node];
        tree[node<<1]=prepre[(mid-l+1)-1]*lazy[node]%mod;
        tree[node<<1|1]=prepre[(r-mid)-1]*lazy[node]%mod;
        lazy[node]=0;
    }
}

void pushup(int node,int l,int r){
    int mid=(l+r)/2;
    // [mid+1,r]
    tree[node]=(tree[node<<1]*pre[r-mid]+tree[node<<1|1])%mod;
}
void build(int node,int l,int r){
    if(l==r){
        tree[node]=s[l]-'0'+1;
        return ;
    }
    int mid=(l+r)/2;
    build(node<<1,l,mid);
    build(node<<1|1,mid+1,r);
    pushup(node,l,r);
}
void update(int node,int L,int R,int val,int l,int r){
    if(L<=l&&r<=R){
        tree[node]=prepre[(r-l+1)-1]*val;
        lazy[node]=val;
        return ;
    }
    updown(node,l,r);
    int mid=(l+r)/2;
    if(mid>=L) update(node<<1,L,R,val,l,mid);
    if(mid<R) update(node<<1|1,L,R,val,mid+1,r);
    pushup(node,l,r);
}
ull query(int node,int L,int R,int l,int r){
    if(R<L) return 0;
    if(L<=l&&r<=R){
        return tree[node];
    }
    updown(node,l,r);
    int mid=(l+r)/2;
    ull ans=0;
    if(mid>=L) ans+= pre[max(0,min(R , r)- mid)]*query(node<<1,L,R,l,mid);
    if(mid<R) ans+=query(node<<1|1,L,R,mid+1,r);
    return ans%mod;
}
int main(){
    cin>>n>>m>>k;
    cin>>s+1;
    pre[0]=1;
    prepre[0]=1;
    for(int i=1;i<=n;i++){
        pre[i]=pre[i-1]*base%mod;
        prepre[i]=(prepre[i-1]+pre[i])%mod;
    }
    build(1,1,n);
    m+=k;
    for(int i=1,opt,l,r,c;i<=m;i++){
        cin>>opt>>l>>r>>c;
        if(opt==1){
            update(1,l,r,c+1,1,n);
        }else{

            ull temp1=query(1,l,r-c,1,n);
            ull temp2=query(1,l+c,r,1,n);
            if(temp1==temp2){
                cout<<"YES\n";
            }else{
                cout<<"NO\n";
            }
        }
    }
    return 0;
}