题意
有n只熊,从1到n进行编号。
第i只熊的电话号码是si。每只熊会给那些电话号码是他的子串的熊打电话(可能会给自己打)。
call(i, j) 表示第i只熊给第j只熊打电话的次数,也就是第j个串在第i个串中出现的次数。
迈克会有q次询问。每个询问中给出l,r,k,然后请您计算一下
∑ri=lcall(i,k)
。
1 ≤ n ≤ 2 × 10^5 , 1 ≤ q ≤ 5 × 10^5,所有串的长度之和≤ 2 × 10^5。
分析
由于已经多年没有碰过后缀数组,所以并没有想到后缀数组的解法。
我们可以把所有串拼在一起,在串之间随意插入一个字符,然后跑出其sa。
对于一个询问(l,r,k),也就是要求出在原串的一段区间内k这个字符串出现了多少次。
如果没有l,r的限制,用后缀数组+二分的经典做法显然可以做到nlogn。
加上限制以后,我们可以对每个位置求出其所在的字符串,然后用主席树来查询区间中[l,r]出现的次数即可。
在卡常的时候很zz的自己写了一个min函数,结果怎么卡都卡不过。果然还是系统自带的函数比较优越。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=400005;
int n,q,b[N],rank[N*2],c[N],d[N],s[N],sa[N],rmq[N][19],posl[N/2],posr[N/2],bin[25],sz,stack[15],l,r,ul,ur;
struct tree{int l,r,s;}t[N*19];
char ch[N/2];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void pri(int x)
{
if (!x) {puts("0");return;}
int top=0;
while (x) stack[++top]=x%10,x/=10;
while (top) putchar(stack[top]+'0'),top--;
puts("");
}
void get_sa(int n,int m)
{
for (int i=1;i<=n;i++) b[s[i]]++;
for (int i=1;i<=m;i++) b[i]+=b[i-1];
for (int i=n;i>=1;i--) c[b[s[i]]--]=i;
int t=0;
for (int i=1;i<=n;i++)
{
t+=(s[c[i]]!=s[c[i-1]]);
rank[c[i]]=t;
}
int j=1;
while (j<=n)
{
for (int i=1;i<=t;i++) b[i]=0;
for (int i=1;i<=n;i++) b[rank[i+j]]++;
for (int i=1;i<=t;i++) b[i]+=b[i-1];
for (int i=n;i>=1;i--) c[b[rank[i+j]]--]=i;
for (int i=1;i<=t;i++) b[i]=0;
for (int i=1;i<=n;i++) b[rank[i]]++;
for (int i=1;i<=t;i++) b[i]+=b[i-1];
for (int i=n;i>=1;i--) d[b[rank[c[i]]]--]=c[i];
t=0;
for (int i=1;i<=n;i++)
{
int x=d[i],y=d[i-1];
t+=(rank[x]!=rank[y]||rank[x]==rank[y]&&rank[x+j]!=rank[y+j]);
c[x]=t;
}
for (int i=1;i<=n;i++) rank[i]=c[i];
if (t==n) break;
j<<=1;
}
for (int i=1;i<=n;i++) sa[rank[i]]=i;
}
void get_height(int n)
{
int k=0;
for (int i=1;i<=n;i++)
{
if (k) k--;
int j=sa[rank[i]-1];
while (i+k<=n&&j+k<=n&&s[i+k]==s[j+k]) k++;
rmq[rank[i]][0]=k;
}
}
void get_rmq(int n)
{
bin[0]=1;
for (int i=1;i<=20;i++) bin[i]=bin[i-1]<<1;
int w=0;
c[0]=-1;
for (int i=1;i<=n;i++)
if (i==bin[w]) c[i]=c[i-1]+1,w++;
else c[i]=c[i-1];
for (int j=1;j<=c[n];j++)
for (int i=1;i+bin[j]-1<=n;i++)
rmq[i][j]=min(rmq[i][j-1],rmq[i+bin[j-1]][j-1]);
}
int get_mn(int l,int r)
{
int w=c[r-l+1];
return min(rmq[l][w],rmq[r-bin[w]+1][w]);
}
void ins(int &d,int p,int x)
{
d=++sz;t[d]=t[p];t[d].s++;
if (l==r) return;
int mid=l+r>>1;
if (x<=mid) r=mid,ins(t[d].l,t[p].l,x);
else l=mid+1,ins(t[d].r,t[p].r,x);
}
int query(int d,int p,int x)
{
if (l==r) return t[d].s-t[p].s;
int mid=l+r>>1;
if (x<=mid) {r=mid;return query(t[d].l,t[p].l,x);}
else {l=mid+1;return query(t[d].r,t[p].r,x)+t[t[d].l].s-t[t[p].l].s;}
}
void get_pts(int n,int w,int len)
{
l=2;r=w;
while (l<=r)
{
int mid=l+r>>1;
if (get_mn(mid,w)>=len) r=mid-1;
else l=mid+1;
}
ul=r+1;
l=w+1;r=n;
while (l<=r)
{
int mid=l+r>>1;
if (get_mn(w+1,mid)>=len) l=mid+1;
else r=mid-1;
}
ur=l-1;
}
int main()
{
n=read();q=read();
int now=0;
for (int i=1;i<=n;i++)
{
if (i>1) s[++now]=30;
gets(ch);
int len=strlen(ch);
posl[i]=now+1;posr[i]=now+len;
for (int j=0;j<len;j++) s[++now]=ch[j]-'a'+1;
}
get_sa(now,30);
get_height(now);
get_rmq(now);
for (int i=1;i<=now;i++) b[i]=0;
for (int i=1;i<=n;i++)
for (int j=posl[i];j<=posr[i];j++)
b[rank[j]]=i;
for (int i=1;i<=now;i++)
if (b[i]) l=1,r=n,ins(d[i],d[i-1],b[i]);
else d[i]=d[i-1];
while (q--)
{
int x=read(),y=read(),k=read();
get_pts(now,rank[posl[k]],posr[k]-posl[k]+1);
l=1,r=n;int ans=query(d[ur],d[ul-2],y);
if (x>1) l=1,r=n,ans-=query(d[ur],d[ul-2],x-1);
pri(ans);
}
return 0;
}