BZOJ1396:识别子串(SAM)

Description

Input

一行,一个由小写字母组成的字符串S,长度不超过10^5

Output

L行,每行一个整数,第i行的数据表示关于S的第i个元素的最短识别子串有多长.

Sample Input

agoodcookcooksgoodfood

Sample Output

1
2
3
3
2
2
3
3
2
2
3
3
2
1
2
3
3
2
1
2
3
4

Solution

1A挺开心的省得调了
对于SAM上的每一个节点，我们只需要考虑right集合大小为1的
设一个right集合大小为1的点结束点在endpos，有效长度为[l,r]
那么对于区间[endpos-r+1,endpos-l+1]，这个点的贡献为endpos-i+1，用一颗线段树维护endpos+1，i最后计算贡献
对于区间[endpos-l+1,endpos],这个点的贡献为l，再开一颗线段树维护l
最后扫一遍单点查询最小值就好了
标记永久化好像非常短还好写= =

Code

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #define N (200000+1000)
 5 using namespace std;
 6 
 7 char s[N];
 8 int Ans,n;
 9 
10 struct SGT
11 {
12     int Segt[N<<1];
13     SGT(){memset(Segt,0x7f,sizeof(Segt));}
14     
15     void Update(int now,int l,int r,int l1,int r1,int k)
16     {
17         if (r<l1 || l>r1) return;
18         if (l1<=l && r<=r1){Segt[now]=min(Segt[now],k);return;}
19         int mid=(l+r)>>1;
20         Update(now<<1,l,mid,l1,r1,k); Update(now<<1|1,mid+1,r,l1,r1,k);
21     }
22     void Query(int now,int l,int r,int x)
23     {
24         Ans=min(Ans,Segt[now]);
25         if (l==r) return;
26         int mid=(l+r)>>1;
27         if (x<=mid) Query(now<<1,l,mid,x);
28         else Query(now<<1|1,mid+1,r,x);
29     }
30 }SGT[2];
31 
32 struct SAM
33 {
34     int fa[N],son[N][28],right[N],step[N],End[N],od[N],wt[N];
35     int p,q,np,nq,last,cnt;
36     SAM(){last=++cnt;}
37     
38     void Insert(int x,int pos)
39     {
40         p=last; last=np=++cnt; step[np]=step[p]+1; right[np]=1; End[np]=pos;
41         while (p && !son[p][x]) son[p][x]=np,p=fa[p];
42         if (!p) fa[np]=1;
43         else
44         {
45             q=son[p][x];
46             if (step[p]+1==step[q]) fa[np]=q;
47             else
48             {
49                 nq=++cnt; step[nq]=step[p]+1;
50                 memcpy(son[nq],son[q],sizeof(son[q]));
51                 fa[nq]=fa[q]; fa[q]=fa[np]=nq;
52                 while (son[p][x]==q) son[p][x]=nq,p=fa[p];
53             }
54         }
55     }
56     void Init()
57     {
58         int len=strlen(s+1);
59         for (int i=1; i<=cnt; ++i) wt[step[i]]++;
60         for (int i=1; i<=len; ++i) wt[i]+=wt[i-1];
61         for (int i=cnt; i>=1; --i) od[wt[step[i]]--]=i;
62         for (int i=cnt; i>=1; --i) right[fa[od[i]]]+=right[od[i]];
63     }
64     void Solve()
65     {
66         for (int i=1; i<=cnt; ++i)
67         if (right[i]==1)
68         {
69             SGT[0].Update(1,1,n,End[i]-step[i]+1,End[i]-step[fa[i]],End[i]+1);
70             SGT[1].Update(1,1,n,End[i]-step[fa[i]],End[i],step[fa[i]]+1);
71         }
72         for (int i=1; i<=n; ++i)
73         {
74             Ans=0x7fffffff;
75             SGT[0].Query(1,1,n,i); Ans-=i;
76             SGT[1].Query(1,1,n,i);
77             printf("%d\n",Ans);
78         }
79     }
80 }SAM;
81 
82 int main()
83 {
84     scanf("%s",s+1);
85     n=strlen(s+1);
86     for (int i=1; i<=n; ++i)
87         SAM.Insert(s[i]-'a',i);
88     SAM.Init();
89     SAM.Solve();
90 }

转载于:https://www.cnblogs.com/refun/p/9370482.html