Description

Input

Output

輸出一行n個整數，第i個整數表示第i個字符串的答案。

Sample Input

Sample Output

HINT

Source

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdlib>
 4 #include<cmath>
 5 #include<algorithm>
 6 #include<cstdio>
 7 #define maxn 200005
 8 using namespace std;
 9 int n,k;
10 long long c[maxn];
11 long long ans[maxn]; 
12 struct data {
 
13     long long son[maxn][26],link[maxn],step[maxn],last,cnt,t[maxn],size[maxn],tp[maxn];
14     data() {last=cnt=1;}
15     void extend(int x,int id) {
16         int p=last,np=last=++cnt;step[np]=step[p]+1;t[np]=id;
17         while(p&&!son[p][x]) son[p][x]=np,p=link[p];
18         if(!p) link[np]=1
 
;
19         else {
20             int q=son[p][x];
21             if(step[q]==step[p]+1) link[np]=q;
22             else {
23                 int nq=++cnt;
24                 memcpy(son[nq],son[q],sizeof(son[q]));
25                 step[nq]=step[p]+1;
26                 link[nq]=link[q];link[np]=link[q]=nq;size[nq]=size[q];tp[nq]=tp[q];
27                 while(p&&son[p][x]==q) son[p][x]=nq,p=link[p];
28             }
29         }
30         for(int i=np;i&&tp[i]!=id;i=link[i]) size[i]++,tp[i]=id;
31     }
32 }sam;
33 int head[maxn],sum;
34 struct edge {
35     int to,next;
36 }e[maxn];
37 void add(int u,int v) {e[sum].to=v;e[sum].next=head[u];head[u]=sum++;}
38 char s[maxn];
39 void dfs(int x) {
40     c[x]+=c[sam.link[x]];ans[sam.t[x]]+=c[x];
41     for(int i=head[x];i>=0;i=e[i].next) dfs(e[i].to);
42 }
43 int main() {
44     memset(head,-1,sizeof(head));
45     scanf("%d%d",&n,&k);
46     for(int i=1;i<=n;i++) {
47         scanf("%s",s+1);
48         int len=strlen(s+1);sam.last=1;
49         for(int j=1;j<=len;j++) sam.extend(s[j]-‘a‘,i);
50     }
51     for(int i=2;i<=sam.cnt;i++)    {
52         add(sam.link[i],i);
53         if(sam.size[i]>=k) c[i]=sam.step[i]-sam.step[sam.link[i]];
54     }
55     dfs(1);
56     for(int i=1;i<=n;i++) printf("%lld ",ans[i]);
57 }
58 /*
59 3 2
60 abaac
61 bba
62 bbacb
63 */

[BZOJ3277]串 廣義後綴自動機