题目大意,给出一个长度为n的数字串,一个常数k,然后是m个询问:从l到r位置,能不能找到一个长度为k的指定数字串。
思路:字符串hash之后,使用可持久化权值线段树查找就ok,字符串hash还是第一次写,即指定一个key,key应该尽量大,代表key进制的数,那么对于一个指定的字符串,我们从左到右进行取数字,将其表示成一个key进制的数,存储下每一位的结果,即sum[i]=sum[i-1]*key+a[i](使用unsigned long long让它自己爆掉,自己取余),那么取出一个以i结尾,长度为k的子串的hash值就是sum[i]-sum[i-k]*(key^k);
附上代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
typedef unsigned long long ll;
using namespace std;
int n,m,k,tot=0;
ll sum[200005]={0},element[200005]={0},key_k=1;
const int key=10016959;
const ll maxn=18446744073709551615ull;
int root[200005]={0};
struct node
{
int ls,rs,sum;
}tree[10000000];
void Insert(int dad,int &son,ll l,ll r,ll pos,int v)
{
son=++tot;
tree[son]=tree[dad];
tree[son].sum+=v;
if(l==r)return;
ll mid=(l>>1)+(r>>1)+(l&r&1);
if(pos<=mid)Insert(tree[dad].ls,tree[son].ls,l,mid,pos,v);
else Insert(tree[dad].rs,tree[son].rs,mid+1,r,pos,v);
}
int Ask(int x,int y,ll l,ll r,ll pos)
{
int sum=tree[y].sum-tree[x].sum;
if(l==r)return sum;
if(!sum)return 0;
ll mid=(l>>1)+(r>>1)+(l&r&1);
if(pos<=mid)return Ask(tree[x].ls,tree[y].ls,l,mid,pos);
else return Ask(tree[x].rs,tree[y].rs,mid+1,r,pos);
}
int main(){
scanf("%d%d%d",&n,&m,&k);
int v;
for(int i=1;i<=n;i++)
{
scanf("%d",&v);
sum[i]=sum[i-1]*key+v;
}
for(int i=1;i<=k;i++)key_k*=key;
root[k-1]=tot++;
tree[tot].ls=tree[tot].rs=tree[tot].sum=0;
for(int i=k;i<=n;i++)Insert(root[i-1],root[i],0ull,maxn,sum[i]-sum[i-k]*key_k,1);
int l,r;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&l,&r);
ll temp=0;
int ans=0;
for(int j=1;j<=k;j++)
{
scanf("%d",&v);
temp=temp*key+v;
}
if(r-l+1<k)ans=0;
else ans=Ask(root[l+k-2],root[r],0ull,maxn,temp);
if(ans==0)puts("Yes");
else puts("No");
}
return 0;
}