不明白这个题为啥会是div1E。。。
套路的,给所有串建个AC自动机,插入的时候给路径打标记,这样统计一个串在
[
l
,
r
]
[l,r]
[l,r]中出现了多少次就等价于统计在这个点fail树的子树中有多少个
[
l
,
r
]
[l,r]
[l,r]间的标记,直接将询问离线后扫描线,对dfs序用树状数组维护即可。
时间复杂度
O
(
(
∑
i
=
1
n
∣
s
i
∣
+
q
)
log
n
)
\mathcal O((\sum_{i=1}^{n}|s_i|+q)\log n)
O((∑i=1n∣si∣+q)logn)。
#include <bits/stdc++.h>
#define end end2
#define lowbit(x) (x&-x)
using namespace std;
int sumv[200005];
void add(int x,int n) {
for(;x<=n;x+=lowbit(x)) sumv[x]++;
}
int sum(int x) {
int s=0;
for(;x;x-=lowbit(x)) s+=sumv[x];
return s;
}
int sum(int l,int r) {
return sum(r)-sum(l-1);
}
namespace AC {
int ch[200005][26],fa[200005],tot=1;
int insert(char *s,int n) {
int x=1;
for(int i=0;i<n;i++) {
int v=s[i]-'a';
if (!ch[x][v]) {
ch[x][v]=++tot;
fa[tot]=x;
}
x=ch[x][v];
}
return x;
}
int fail[200005];
queue <int> q;
void bfs() {
for(int i=0;i<26;i++)
if (ch[1][i]) {
fail[ch[1][i]]=1;
q.push(ch[1][i]);
}
else ch[1][i]=1;
while (!q.empty()) {
int x=q.front();q.pop();
for(int i=0;i<26;i++)
if (ch[x][i]) {
fail[ch[x][i]]=ch[fail[x]][i];
q.push(ch[x][i]);
}
else ch[x][i]=ch[fail[x]][i];
}
}
int dfn[200005],end[200005],dfs_cnt;
vector <int> son[200005];
void dfs(int x) {
dfn[x]=++dfs_cnt;
for(int i=0;i<son[x].size();i++)
dfs(son[x][i]);
end[x]=dfs_cnt;
}
void build() {
bfs();
for(int i=2;i<=tot;i++) son[fail[i]].push_back(i);
dfs(1);
}
}
int ed[200005];
struct Query {
int x,k,v,id;
Query() {}
Query(int a,int b,int c,int d):x(a),k(b),v(c),id(d) {}
bool operator < (const Query & b) const {return x<b.x;}
};
Query q[1000005];
int ans[500005];
char str[200005];
int main() {
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%s",str);
int len=strlen(str);
ed[i]=AC::insert(str,len);
}
AC::build();
int cnt=0;
for(int i=1;i<=m;i++) {
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
q[++cnt]=Query(y,z,1,i);
if (x>1) q[++cnt]=Query(x-1,z,-1,i);
}
sort(q+1,q+cnt+1);
int r=1;
for(int i=1;i<=n;i++) {
for(int j=ed[i];j;j=AC::fa[j]) add(AC::dfn[j],AC::tot);
while (r<=cnt&&q[r].x<=i) {
int x=ed[q[r].k];
ans[q[r].id]+=sum(AC::dfn[x],AC::end[x])*q[r].v;
r++;
}
}
for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
return 0;
}