線段樹+哈希【CF580E】Kefa and Watch

線段樹+哈希【CF580E】Kefa and Watch

Description

\(n\)個數的字符串，\(m + k\)個操做

1 l r k把\(l - r\)賦值爲\(k\)

2 l r d詢問\(l - r\)是否有長度爲\(d\)的循環節

\(n \leq 10^5, m + k \leq 10^5, d \leq 10\)

Input

第一行爲三個整數\(n,m,k\)

第二行爲一個\(n\)個數的字符串。

接下來\(m+k\)行每行對應一種操做。

Output

對於每個\(2\)操做,若是存在,輸出一行\(YES\)，不然輸出\(NO\)

線段樹維護哈希

寫起來爽,調起來更爽

咱們首先預處理出\(po\)數組記錄\(base^i\)(這個要用來修改及查詢的。)

還要預處理出來\(val[i][j]\)表明長度爲\(j\)的所有爲數字\(i\)的字符串的哈希值。

而後每次區間合併的時候.
\[ len=tr[rs].r-tr[rs].l+1 \\ tr[o].va=(tr[ls].va\times po[len]％\ mod +tr[rs].va) ％\ mod \]
這個應該不是很難理解吧。(就相似於你\(hash\)匹配的作法。)

修改時候,咱們直接賦值\(tr[o].va=val[k][len]\)便可。

須要注意的有兩點：

1. \(lazy\)標記初值要爲\(1\),由於會存在賦值爲\(0\)的狀況
2. 查詢操做中,當前區間分別在左右兩側的時候\(tr[ls].va \times po[r-mid]\)！！

所以直接碼代碼就行了

還有一個神仙結論是作題的根據。

若是詢問爲\((l,r,d)\),則只須要判斷\((l+d,r)\)和\((l,r-d)\)便可。

證實的話,我不太會.可是這是正確的。

若是這題卡單\(hash\)的話能夠寫雙\(hash\)。稍做修改便可。很少\(BB\)了.

代碼

#include<cstdio>
#include<algorithm>
#include<iostream>
#define lo long long
#define base 31
#define mod 20020303
#define R register

using namespace std;

const int gz=1e5+8;

inline void in(R int &x)
{
    int f=1;x=0;char s=getchar();
    while(!isdigit(s)){if(s=='-')f=-1;s=getchar();}
    while(isdigit(s)){x=x*10+s-'0';s=getchar();}
    x*=f;
}

int n,m,K,po[gz]={1},val[10][gz];

char s[gz];

struct wc
{
    int l,r,tg;
    lo va;
}tr[gz<<2];

inline void pre()
{
    for(R int i=1;i<gz;i++)
        po[i]=po[i-1]*base%mod;
    for(R int i=0;i<10;i++)
        for(R int j=1;j<gz;j++)
            val[i][j]=(val[i][j-1]*base%mod+i)%mod;
}

#define ls o<<1
#define rs o<<1|1

inline void up(R int o)
{
    tr[o].va=(tr[ls].va*po[tr[rs].r-tr[rs].l+1]%mod+tr[rs].va%mod)%mod;
}

void build(R int o,R int l,R int r)
{
    tr[o].l=l,tr[o].r=r;tr[o].tg=-1;
    if(l==r)
    {
        tr[o].va=s[l]-'0';
        return;
    }
    R int mid=(l+r)>>1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    up(o);
}

inline void down(R int o)
{
    if(tr[o].tg==-1)return;
    R int k=tr[o].tg;
    tr[ls].va=val[k][tr[ls].r-tr[ls].l+1];
    tr[rs].va=val[k][tr[rs].r-tr[rs].l+1];
    tr[ls].tg=tr[rs].tg=k;
    tr[o].tg=-1;
}

void change(R int o,R int l,R int r,R int k)
{
    if(tr[o].l==l and tr[o].r==r)
    {
        tr[o].tg=k;
        tr[o].va=val[k][tr[o].r-tr[o].l+1];
        return ;
    }
    down(o);
    R int mid=(tr[o].l+tr[o].r)>>1;
    if(r<=mid)change(ls,l,r,k);
    else if(l>mid)change(rs,l,r,k);
    else change(ls,l,mid,k),change(rs,mid+1,r,k);
    up(o);
}

lo query(R int o,R int l,R int r)
{
    if(tr[o].l==l and tr[o].r==r)return tr[o].va;
    down(o);
    R int mid=(tr[o].l+tr[o].r)>>1;
    if(r<=mid)return query(ls,l,r);
    else if(l>mid) return query(rs,l,r);
    else
        return ((query(ls,l,mid)%mod)*po[r-mid]%mod+query(rs,mid+1,r)%mod)%mod;//注意這裏！！
}

int main()
{
    pre();
    in(n),in(m),in(K);
    R int tt=m+K;
    scanf("%s",s+1);
    build(1,1,n);
    for(R int opt,l,r,k;tt;tt--)
    {
        in(opt),in(l),in(r),in(k);
        switch(opt)
        {
            case 1:change(1,l,r,k);break;
            case 2:
            {
                if(r-l+1==k)
                {
                    puts("YES");
                    continue;
                }
                puts(query(1,l,r-k)==query(1,l+k,r) ? "YES":"NO");
                break;
            }
        }
    }
}