1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #define M 1000008 5 using namespace std; 6 char ch[M]; 7 int a[M][27],n,cnt=1,fa[M],f1[M],ans[M],q[M]; 8 long long sum[M],aa; 9 void jia() 10 { 11 int now=1; 12 for(int i=1;i<=strlen(ch+1);i++) 13 if(!a[now][ch[i]-96]) 14 { 15 cnt++; 16 a[now][ch[i]-96]=cnt; 17 sum[cnt]++; 18 now=cnt; 19 } 20 else 21 { 22 now=a[now][ch[i]-96]; 23 sum[now]++; 24 } 25 ans[0]++; 26 ans[ans[0]]=now; 27 } 28 void build() 29 { 30 int h=0,t=1; 31 q[1]=1; 32 for(;h<t;) 33 { 34 int p=q[++h]; 35 for(int i=1;i<=26;i++) 36 if(a[p][i]) 37 { 38 int now=fa[p]; 39 q[++t]=a[p][i]; 40 for(;!a[now][i];now=fa[now]); 41 fa[a[p][i]]=a[now][i]; 42 } 43 } 44 return; 45 } 46 void dfs(int a1) 47 { 48 f1[a1]=1; 49 int now; 50 for( now=fa[a1];f1[now]&&now>1;sum[now]+=sum[a1],now=fa[now]); 51 sum[now]+=sum[a1]; 52 for(int i=1;i<=26;i++) 53 if(a[a1][i]) 54 dfs(a[a1][i]); 55 return; 56 } 57 int main() 58 { 59 scanf("%d",&n); 60 for(int i=1;i<=n;i++) 61 { 62 scanf("%s",ch+1); 63 jia(); 64 } 65 for(int i=1;i<=26;i++) 66 a[0][i]=1; 67 build(); 68 dfs(1); 69 for(int i=1;i<=n;i++) 70 printf("%lld\n",sum[ans[i]]); 71 return 0; 72 }
AC自动机
