题目链接:https://www.luogu.org/problemnew/show/P3808
要注意的是一定要把len赋值strlen(s);不然超时超的自闭!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
#include <bits/stdc++.h> using namespace std; #define ll long long #define re register #define fi first #define se second #define mp make_pair #define pb push_back #define P pair<int,int> const int N=1e6+10; void read(int &a) { a=0; int d=1; char ch; while(ch=getchar(),ch>'9'||ch<'0') if(ch=='-') d=-1; a=ch-'0'; while(ch=getchar(),ch>='0'&&ch<='9') a=a*10+ch-'0'; a*=d; } void write(int x) { if(x<0) putchar(45),x=-x; if(x>9) write(x/10); putchar(x%10+'0'); } struct note { int to[30],bz,nex; }trie[N]; int tot; char s[N]; void ins(char *s) { int t=0; int len=strlen(s); for(re int i=0;i<len;i++) { int x=s[i]-96; if(!trie[t].to[x]) trie[t].to[x]=++tot; t=trie[t].to[x]; } trie[t].bz++; } void built() { queue <int> q; for(re int i=1;i<=26;i++) if(trie[0].to[i]) trie[trie[0].to[i]].nex=0,q.push(trie[0].to[i]); while(!q.empty()) { int p=q.front(); q.pop(); for(re int i=1;i<=26;i++) if(trie[p].to[i]) trie[trie[p].to[i]].nex=trie[trie[p].nex].to[i],q.push(trie[p].to[i]); else trie[p].to[i]=trie[trie[p].nex].to[i]; } } int solve(char *s) { int ans=0,now=0; int len=strlen(s); for(re int i=0;i<len;i++) { now=trie[now].to[s[i]-'a'+1]; for(re int j=now;j&&trie[j].bz!=-1;j=trie[j].nex) ans+=trie[j].bz,trie[j].bz=-1; } return ans; } int main() { int n; read(n); while(n--) { scanf("%s",s); ins(s); } trie[0].nex=0; built(); scanf("%s",s); cout<<solve(s)<<endl; return 0; }