Description
给出n个字符串,这n个字符串的总长为L,定义函数call(i,j)表示第j个字符串在第i个字符串中出现的次数。
给出q次询问,每次询问给出l,r,k,求
∑ri=lcall(i,k)
n,l<=2*1e5,q<=5*1e5
Solution
蒟蒻不会fail树做法QwQ
只会最朴素的SA啊_ (:з」∠) _
首先把所有串接在一起建SA
那么对于每个串k,它作为子串的串在SA中一定是一段区间。
那么我们可以用二分找出这个区间,判断条件就是lcp是否大于串长
接下来用主席树维护这个区间中有多少个在[l,r]内的串。
卡常卡的欲仙欲死FAQ
18次提交还好51nod不算正确率23333
辣鸡卡常nod
Code
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
inline int read() {
char ch;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
int x=ch-'0';
for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
inline void write(int x) {
if (!x) {puts("0");return;}
char ch[20];int tot=0;
for(;x;x/=10) ch[++tot]=x%10+'0';
fd(i,tot,1) putchar(ch[i]);
puts("");
}
#define min(x,y) (x<y?x:y)
#define max(x,y) (x<y?y:x)
const int N=4*1e5+5;
int L,w[N],x[N],y[N],sa[N],rank[N],height[N];
char s[N];
inline void tsort() {
int mx=0;
fo(i,1,L) w[x[y[i]]]++,mx=max(mx,x[y[i]]);
fo(i,1,mx) w[i]=w[i]+w[i-1];
fd(i,L,1) sa[w[x[y[i]]]--]=y[i];
fo(i,1,mx) w[i]=0;
}
inline void get_sa() {
tsort();
for(int j=1;j<=L;j<<=1) {
int k=0;
fo(i,L-j+1,L) y[++k]=i;
fo(i,1,L) if (sa[i]>j) y[++k]=sa[i]-j;
tsort();
fo(i,1,L) y[i]=x[i],x[i]=0;
x[sa[1]]=k=1;
fo(i,2,L) {
if (y[sa[i]]!=y[sa[i-1]]||y[sa[i]+j]!=y[sa[i-1]+j]) k++;
x[sa[i]]=k;
}
if (k==L) break;
}
fo(i,1,L) rank[sa[i]]=i;
}
inline void get_height() {
int k=0;
fo(i,1,L) {
if (k) k--;
int j=sa[rank[i]-1];
while (i+k<=L&&j+k<=L&&s[i+k]==s[j+k]) k++;
height[rank[i]]=k;
}
}
int tot,ql,qr,k,sum,root[N],le[N<<4],ri[N<<4],tr[N<<4];
inline void insert(int &v,int l,int r,int x) {
++tot;
le[tot]=le[v];ri[tot]=ri[v];
tr[tot]=tr[v]+1;
v=tot;
if (l==r) return;
int m=l+r>>1;
if (x<=m) insert(le[v],l,m,x);
else insert(ri[v],m+1,r,x);
}
inline void query(int v,int l,int r) {
if (!v) return;
if (ql<=l&&r<=qr) {sum=sum+tr[v];return;}
int m=l+r>>1;
if (m>=ql) query(le[v],l,m);
if (m<qr) query(ri[v],m+1,r);
}
int n,q,len[N],pos[N],id[N];
char st[N];
inline void init() {
n=read();q=read();
fo(i,1,n) {
char ch;pos[i]=++L;len[i]=1;
for(ch=getchar();ch<'a'||ch>'z';ch=getchar());
s[L]=ch;id[L]=i;
for(ch=getchar();ch>='a'&&ch<='z';ch=getchar())
++len[i],id[++L]=i,s[L]=ch;
s[++L]='a'+26;
}
fo(i,1,L) x[i]=s[i]-'a'+1;
fo(i,1,L) y[i]=i;
get_sa();get_height();
}
int f[19][N],two[19],lg[N];
inline int lcp(int x,int y) {
if (x==y) return len[id[sa[x]]];
x++;int z=lg[y-x+1];
return min(f[z][x],f[z][y-two[z]+1]);
}
inline void prepare() {
two[0]=1;fo(i,1,18) two[i]=two[i-1]<<1;
fo(i,2,L) f[0][i]=height[i];
fo(i,1,L) lg[i]=log(i)/log(2);
fo(j,1,18)
fo(i,2,L-two[j]+1)
f[j][i]=min(f[j-1][i],f[j-1][i+two[j-1]]);
fo(i,1,L) {
root[i]=root[i-1];
if (id[sa[i]]) insert(root[i],1,n,id[sa[i]]);
}
}
int lx[N],rx[N];
inline int find_l(int l,int r,int k) {
int now=r;
while (l<r) {
int mid=l+r>>1;
if (lcp(mid,now)>=k) r=mid;
else l=mid+1;
}
return l;
}
inline int find_r(int l,int r,int k) {
int now=l;
while (l<r) {
int mid=l+r>>1;
if (lcp(now,mid)>=k) l=mid+1;
else r=mid;
}
if (lcp(now,l)<k) l--;
return l;
}
inline void solve() {
fo(i,1,n) {
lx[i]=find_l(1,rank[pos[i]],len[i]);
rx[i]=find_r(rank[pos[i]],L+1,len[i]);
}
for(;q;q--) {
ql=read();qr=read();k=read();
int lpos=lx[k],rpos=rx[k];
int ans=0;
sum=0;query(root[rpos],1,n);ans=ans+sum;
sum=0;query(root[lpos-1],1,n);ans=ans-sum;
write(ans);
}
}
int main() {
init();
prepare();
solve();
return 0;
}