我通过下载 2 号测试点的样例,输出与 out
时一模一样的,均为 15874。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e6;
const int M=2e6;
int n;
int ac[N+5][30],tot;
int nxt[N+5];
char s[N+5];
int cnt[N+5];
void insert(char a[N+5]){
int n=strlen(a+1);
int u=0;
for(int i=1;i<=n;i++){
if(!ac[u][a[i]-'a'])
ac[u][a[i]-'a']=++tot;
u=ac[u][a[i]-'a'];
}
++cnt[u];
return ;
}
void build(){
queue<int> q;
for(int i=0;i<26;i++)
if(ac[0][i])
q.push(ac[0][i]);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=0;i<26;i++){
int v=ac[u][i];
if(v)
nxt[v]=ac[nxt[u]][i],q.push(v);
else
ac[u][i]=ac[nxt[u]][i];
}
}
return ;
}
int ask(char t[M+5]){
int res=0;
int u=0;
int n=strlen(s+1);
for(int i=1;i<=n;i++){
int u=ac[u][t[i]-'a'];
for(int j=u;j&&cnt[j]!=-1;j=nxt[j]){
res+=cnt[j];
cnt[j]=-1;
}
}
return res;
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
cin>>s+1;
insert(s);
}
build();
cin>>s+1;
cout<<ask(s)<<"\n";
return 0;
}