Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 21842 Accepted Submission(s): 6968
#include<iostream> #include<string.h> #include<string> #include<algorithm> #include<queue> #define N 2000006 using namespace std; int tree[N][128],vis[N],ans[N],last[N],fail[N]; int n,id,root,num=0; string s[1010],ss; void insert(int x) { root=0; for(int i=0;s[x][i];i++) { id=s[x][i]; if(tree[root][id]==0) { tree[root][id]=++num; memset(tree[num],0,sizeof(tree[num])); vis[num]=0; } root=tree[root][id]; } vis[root]=x; } void build() { queue<int>p; fail[0]=0; for(int i=0;i<128;i++) { if(tree[0][i]!=0) { fail[tree[0][i]]=0; last[tree[0][i]]=0; p.push(tree[0][i]); } } while(!p.empty()) { root=p.front(); p.pop(); for(int i=0;i<128;i++) { if(tree[root][i]==0) { tree[root][i]=tree[fail[root]][i]; continue; } p.push(tree[root][i]); int temp=tree[root][i]; int fa=fail[root]; fail[temp]=tree[fa][i]; last[temp]=(vis[fail[temp]]?fail[temp]:last[fail[temp]]); } } } void count(int x) { if(x!=0) { ans[vis[x]]++; count(last[x]); } } void search() { root=0; for(int i=0;ss[i];i++) { id=ss[i]; while(root&&tree[root][id]==0) root=fail[root]; root=tree[root][id]; if(vis[root]) count(root); else if(last[root]) count(last[root]); } } int main() { while(cin>>n) { num=0; memset(tree[0],0,sizeof(tree[0])); memset(ans,0,sizeof(ans)); fail[0]=vis[0]=last[0]; for(int i=1;i<=n;i++) { cin>>s[i]; insert(i); } build(); ss.clear(); cin>>ss; search(); for(int i=1;i<=n;i++) { if(ans[i]!=0) cout<<s[i]<<": "<<ans[i]<<endl; } } return 0; }
原文:https://www.cnblogs.com/-citywall123/p/11319272.html