识别子串 (string)——后缀自动机+线段树

题目

【题目描述】

一般地，对于一个字符串 S，和 S 中第 $ i $ 个字符 x，定义子串 $ T=S(i.j) $ 为一个关于 x 的识别子申，当且仅当：
　　1.$ i \leq x \leq j $
　　2.T 在 S 巾只出现一次
比如，对于 banana 的第 $ 5 $ 个字符，“nana”, “anan”，“anana”, “nan”，“banan” 和“banana”都是关于它的识别子串。
说你写一个程序,计算出对对于一个字符串 S，关于 S 的每一位的最短识别子串的长度。

【输入格式】

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

【输出格式】

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

【样例输入】

agoodcookcooksgoodfood

【样例输出】

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

【数据范围与提示】

对于 20% 的数据 L<=200

对于 40% 的数据 L<=5000

对于 60% 的数据 L<=20000

对于 100% 的数据 L<=100000

题解

可以发现，对于每一个唯一识别的子串，都可以更新一段答案

可以发现，在后缀自动机上的每一个叶子节点 $ x $，都是一段唯一识别的子串

考虑更新，记 $ l=len(x)-len(fa(x)),r=len(x) $，那么对于 $ [1,l-1] $ 的贡献为 $ r-i+1 $，对于 $ [l,r] $ 的贡献为 $ r-l+1 $

可以把 $r-i+1$ 的 $ i $ 提出，转化为 $query(i)-i$，开两棵线段树分别统计即可

代码

 #include<bits/stdc++.h>
 #define LL long long
 #define _(d) while(d(isdigit(ch=getchar())))
 using namespace std;
 int R(){
     int x;bool f=;char ch;_(!)if(ch=='-')f=;x=ch^;
     _()x=(x<<)+(x<<)+(ch^);return f?x:-x;}
 const int N=1e6+;
 int n,las=,cnt=,Rt=;
 char s[N];bool f[N];
 struct node{int ch[],fa,len;}tr[N];
 void extend(int c){
     int p=las,np=las=++cnt,q,nq;
     tr[np].len=tr[p].len+;
     while(!tr[p].ch[c]&&p)
         tr[p].ch[c]=np,p=tr[p].fa;
     if(!p)return void(tr[np].fa=Rt);
     if(tr[p].len+==tr[q=tr[p].ch[c]].len)return void(tr[np].fa=q);
     tr[nq=++cnt].len=tr[p].len+;
     memcpy(tr[nq].ch,tr[q].ch,sizeof tr[q].ch);
     tr[nq].fa=tr[q].fa,tr[np].fa=tr[q].fa=nq;
     while(p&&tr[p].ch[c]==q)
         tr[p].ch[c]=nq,p=tr[p].fa;
     return;
 }
 class seg{
 private:
     #define Ls rt<<1
     #define Rs rt<<1|1
 public:
     int tr[N];
     seg(){memset(tr,0x3f,sizeof tr);}
     void update(int rt,int l,int r,int ql,int qr,int v){
         if(ql>qr)return;
         if(ql<=l&&qr>=r)return void(tr[rt]=min(tr[rt],v));
         int mid=l+r>>;
         if(ql<=mid)update(Ls,l,mid,ql,qr,v);
         if(qr>mid)update(Rs,mid+,r,ql,qr,v);
         return;
     }
     int query(int rt,int l,int r,int k){
         if(l==r)return tr[rt];
         int mid=l+r>>;
         if(k<=mid)return min(tr[rt],query(Ls,l,mid,k));
         else return min(tr[rt],query(Rs,mid+,r,k));
     }
 }T1,T2;
 int main(){
     scanf("%s",s+),n=strlen(s+);
     for(int i=;i<=n;i++)extend(s[i]-'a');
     for(int i=;i<=cnt;i++)f[tr[i].fa]=;
     for(int i=;i<=cnt;i++)
         if(!f[i]){
             int x=tr[i].len-tr[tr[i].fa].len,y=tr[i].len;
             T1.update(,,n,,x-,y+),T2.update(,,n,x,y,tr[tr[i].fa].len+);
         }
     for(int i=;i<=n;i++)
         printf("%d\n",min(T1.query(,,n,i)-i,T2.query(,,n,i)));
     return ;
 }