bzoj 2865 字符串识别——后缀数组

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=2865

做出 ht[ ] 之后，sa[ ] 上每个位置和它前面与后面取 LCP ，其中较大的长度设为 d ，表示从 sa[ i ] 位置开始的子串的右端点要在 sa[ i ]+d-1 位置之后才是只出现了一次的。

那么 sa[ i ] ~ sa[ i ]+d 位置的答案可以对 d+1 取 min ；至于 sa[ i ]+d+1 ~ n 位置，sa[ i ]可能成为它们答案的开头位置，所以可以维护每个位置备选答案串开头的最靠后位置（这样最靠近自己），让 sa[ i ]+d+1 ~ n 位置的这个值对 sa[ i ] 取 max 就行了。可以用线段树维护。

注意那个“备选答案串开头的最靠后位置”的初值不是0！不然第一个位置有可能和第0个位置组合了。应该是 -n 之类，才能排除影响。

#include<cstdio>

#include<cstring>

#include<algorithm>

#define ls Ls[cr]

#define rs Rs[cr]

using namespace std;

const int N=5e5+,M=N<<;

int n,sa[N],rk[N],tp[N],tx[N],ht[N];

int tot,Ls[M],Rs[M],tg1[M],tg2[M];

int ans[N]; char s[N];

int Mx(int a,int b){return a>b?a:b;}

int Mn(int a,int b){return a<b?a:b;}

void Rsort(int n,int nm)

{

  for(int i=;i<=nm;i++)tx[i]=;

  for(int i=;i<=n;i++)tx[rk[i]]++;

  for(int i=;i<=nm;i++)tx[i]+=tx[i-];

  for(int i=n;i;i--)sa[tx[rk[tp[i]]]--]=tp[i];

}

void get_sa(int n)

{

  int nm=;

  for(int i=;i<=n;i++)tp[i]=i,rk[i]=s[i];

  Rsort(n,nm);

  for(int k=;k<=n;k<<=)

    {

      int tot=;

      for(int i=n-k+;i<=n;i++)tp[++tot]=i;

      for(int i=;i<=n;i++)

    if(sa[i]>k)tp[++tot]=sa[i]-k;

      Rsort(n,nm);memcpy(tp,rk,sizeof rk);

      nm=;rk[sa[]]=;

      for(int i=,u,v;i<=n;i++)

    {

      u=sa[i]+k;v=sa[i-]+k;if(u>n)u=;if(v>n)v=;

      rk[sa[i]]=(tp[sa[i]]==tp[sa[i-]]&&tp[u]==tp[v])?nm:++nm;

    }

      if(nm==n)break;

    }

}

void get_ht(int n)

{

  for(int i=,k=,j;i<=n;i++)

    {

      for(k?k--:,j=sa[rk[i]-];i+k<=n&&j+k<=n&&s[i+k]==s[j+k];k++);

      ht[rk[i]]=k;//rk[i]!!!

    }

}

void build(int l,int r,int cr)

{

  tg1[cr]=n+; tg2[cr]=-n;////

  if(l==r)return; int mid=l+r>>;

  ls=++tot; build(l,mid,ls);

  rs=++tot; build(mid+,r,rs);

}

void mdfy(int l,int r,int cr,int L,int R,int k)

{

  if(l>=L&&r<=R){tg1[cr]=Mn(tg1[cr],k);return;}

  int mid=l+r>>;

  if(L<=mid)mdfy(l,mid,ls,L,R,k);

  if(mid<R)mdfy(mid+,r,rs,L,R,k);

}

void mdfyx(int l,int r,int cr,int L,int R,int k)

{

  if(l>=L&&r<=R){tg2[cr]=Mx(tg2[cr],k);return;}

  int mid=l+r>>;

  if(L<=mid)mdfyx(l,mid,ls,L,R,k);

  if(mid<R)mdfyx(mid+,r,rs,L,R,k);

}

void dfs(int l,int r,int cr,int lj1,int lj2)

{

  lj1=Mn(lj1,tg1[cr]); lj2=Mx(lj2,tg2[cr]);//before l==r

  if(l==r){ans[l]=Mn(lj1,l-lj2+);return;}

  int mid=l+r>>;

  dfs(l,mid,ls,lj1,lj2); dfs(mid+,r,rs,lj1,lj2);

}

int main()

{

  scanf("%s",s+);n=strlen(s+);

  get_sa(n); get_ht(n);

  tot=;build(,n,);

  for(int i=,d;i<=n;i++)

    {

      d=sa[i]+Mx(ht[i],ht[i+]); if(d>n)continue;

      mdfy(,n,,sa[i],d,d-sa[i]+);

      if(d<n)mdfyx(,n,,d+,n,sa[i]);

    }

  dfs(,n,,n+,-n);//-n

  for(int i=;i<=n;i++)printf("%d\n",ans[i]);

  return ;

}