【BZOJ3207】花神的嘲諷計劃Ⅰ hash+可持久化線段樹

空灰冰魂發表於2015-05-13

連結:

#include <stdio.h>
int main()
{
    puts("轉載請註明出處[vmurder]謝謝");
    puts("網址:blog.csdn.net/vmurder/article/details/45688093");
}

題解:

首先因為嘲諷長度固定,所以我們可以給每個點固定一個hash值(不固定的話我還真不會做)。
然後用可持久化線段樹實現一段區間內有哪些數,然後查詢一段區間是否有要的那個數就行了。

程式碼:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 401000
#define LOGN 20
#define base 107
#define inf 0x3f3f3f3f
using namespace std;
int n,m,p;
long long a[N],t[N],h[N];
struct LSH
{
    long long x;
    int i,f;
    bool operator < (const LSH &A)const
    {return x<A.x;}
}lsh[N];
int root[N];
struct Segment_Tree
{
    int l,r,x;
}s[N*LOGN];
int cnt;
void add(int last,int &now,int l,int r,int x)
{
    now=++cnt;
    if(l==r)s[now].x=s[last].x+1;
    else {
        int mid=l+r>>1;
        if(x<=mid)
        {
            add(s[last].l,s[now].l,l,mid,x);
            s[now].r=s[last].r;
        }
        else {
            s[now].l=s[last].l;
            add(s[last].r,s[now].r,mid+1,r,x);
        }
        s[now].x=s[s[now].l].x+s[s[now].r].x;
    }   
}
bool ask(int last,int now,int l,int r,long long x)
{
    if(s[now].x-s[last].x==0)return 0;
    if(lsh[r].x<x)return 0;
    if(lsh[l].x>x)return 0;

    if(lsh[l].x==x&&l==r)return 1;
    int mid=l+r>>1;
    if(x<=lsh[mid].x)return ask(s[last].l,s[now].l,l,mid,x);
    else return ask(s[last].r,s[now].r,mid+1,r,x);
}
int main()
{
    int i,j,k;
    int a,b,c;
    long long baser;
    scanf("%d%d%d",&n,&m,&p);
    for(i=1;i<=n;i++)scanf("%lld",&h[i]),h[i]+=h[i-1]*base;
    for(baser=i=1;i<=p;i++)baser*=base;
    for(i=n;i>=p;i--)h[i]-=h[i-p]*baser;
    for(i=p;i<=n;i++)lsh[i].x=h[i],lsh[i].i=i;
    sort(lsh+p,lsh+n+1);
    for(i=p;i<=n;i++)
    {
        if(i==p||lsh[i].x!=lsh[i-1].x)lsh[++cnt].x=lsh[i].x;
        h[lsh[i].i]=cnt;
    }
    for(k=cnt,cnt=0,i=p;i<=n;i++)add(root[i-1],root[i],1,k,h[i]);
    while(m--)
    {
        scanf("%d%d",&a,&b);

        for(baser=0,i=1;i<=p;i++)
        {
            scanf("%d",&c);
            baser=baser*base+c;
        }
        if(b-a<p)puts("Yes");
        else if(ask(root[a+p-2],root[b],1,k,baser))puts("No");
        else puts("Yes");
    }
    return 0;
}

相關文章