Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 12785 Accepted Submission(s): 4408
//模板题。用val数组记录单词序号,ans数组记录出现次数。 #include<iostream> #include<cstdio> #include<cstring> #include<queue> using namespace std; const int maxnod=2000006; const int sigma_size=128; int nod[maxnod][sigma_size],f[maxnod],val[maxnod],last[maxnod]; int sz,ans[1003]; void init() { sz=1; memset(nod[0],0,sizeof(nod[0])); memset(ans,0,sizeof(ans)); f[0]=val[0]=last[0]=0; } void Insert(char* s,int cnt) { int len=strlen(s); int rt=0; for(int i=0;i<len;i++){ int id=s[i]; if(nod[rt][id]==0){ memset(nod[sz],0,sizeof(nod[sz])); nod[rt][id]=sz; val[sz++]=0; } rt=nod[rt][id]; } val[rt]=cnt; } void getfail() { queue<int>q; f[0]=0; for(int i=0;i<sigma_size;i++){ int u=nod[0][i]; if(u) {f[u]=0;q.push(u);last[u]=0;} } while(!q.empty()){ int r=q.front();q.pop(); for(int i=0;i<sigma_size;i++){ int u=nod[r][i]; if(!u) continue; q.push(u); int v=f[r]; while(v&&!nod[v][i]) v=f[v]; f[u]=nod[v][i]; last[u]=(val[f[u]]?f[u]:last[f[u]]); } } } void print(int i) { if(i){ ans[val[i]]++; print(last[i]); } } void find(char* t) { int len=strlen(t); int j=0; for(int i=0;i<len;i++){ int id=t[i]; while(j&&!nod[j][id]) j=f[j]; j=nod[j][id]; if(val[j]) print(j); else if(last[j]) print(last[j]); } } int main() { int n; char s[1003][60],t[2000006]; while(scanf("%d",&n)==1){ init(); for(int i=1;i<=n;i++){ scanf("%s",s[i]); Insert(s[i],i); } getfail(); scanf("%s",t); find(t); for(int i=1;i<=n;i++){ if(ans[i]) printf("%s: %d\n",s[i],ans[i]); } } return 0; }
原文:http://www.cnblogs.com/--ZHIYUAN/p/6476062.html