poj 3415 Common Substrings——后缀数组+单调栈

题目：http://poj.org/problem?id=3415

因为求 LCP 是后缀数组的 ht[ ] 上的一段取 min ，所以考虑算出 ht[ ] 之后枚举每个位置作为右端的贡献。

一开始想的是把两个数组接起来（中间加个逗号之类的，就能算出正确的 LCP ），不加区分地算了贡献之后再分别减去两个数组自己内部的贡献。

看看题解，得知可以在那个接起来的数组上分别算 a 与前面的 b 、b 与前面的 a 的贡献，就不用容斥了。

考虑怎么算贡献。一开始想的是取 min 一定越取越小，所以维护双指针卡在取min>=K的最靠前位置；新加入一个元素的时候二分找到第一个取min后比自己小的位置，改一下该位置到自己位置的贡献（现在看来好像不用维护那个双指针也行？）；

看看题解，原来可以用单调栈。就相当于维护一下单调栈的“面积”一样。

不过有一点麻烦的是 i 位置能否产生贡献是看 i+1 位置到末尾位置的 min ；这样求“面积”就要费点心；不过想到可以让 ht[ i ] = ht[ i+1 ] ，这样的话正常求面积就行了！累计到答案上之后再把当前位置入队。

注意有大写字母。而且 poj 上编译不过 swap( rk , tp ) ，写 memcpy 就行了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+,M=N<<;
int n,m,s[M],sa[M],rk[M],tp[M],tx[M],ht[M];
int sta[M],top,wd1[M],wd2[M]; ll sm1[M],sm2[M];
char a[N],b[N];
int Mx(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);
      /*swap(rk,tp);*/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)
{
  int k=,j;
  for(int i=;i<=n;i++)
    {
      for(j=sa[rk[i]-],k?k--:;i+k<=n&&j+k<=n&&s[i+k]==s[j+k];k++);
      ht[rk[i]]=k;
    }
}
int main()
{
  while()
    {
      int lm;scanf("%d",&lm);if(!lm)return ;
      scanf("%s",a+);scanf("%s",b+);
      n=strlen(a+); m=strlen(b+); int tn=n+m+;
      for(int i=;i<=n;i++)s[i]=a[i]-'A'+;
      s[n+]=;
      for(int i=n+,j=;i<=tn;i++,j++)s[i]=b[j]-'A'+;
      get_sa(tn);get_ht(tn);

      for(int i=;i<tn;i++)ht[i]=ht[i+];
      top=; ll ans=;
      for(int i=;i<=tn;i++)
    {
      if(sa[i]<=n)ans+=sm2[top]; else if(sa[i]>n+)ans+=sm1[top];
      int lj1=,lj2=;
      while(top&&ht[i]<=ht[sta[top]])
        lj1+=wd1[top],lj2+=wd2[top],top--;
      sta[++top]=i;
      if(sa[i]<=n)lj1++; else if(sa[i]>n+)lj2++;
      wd1[top]=lj1; sm1[top]=sm1[top-]+(ll)lj1*Mx(,ht[i]-lm+);
      wd2[top]=lj2; sm2[top]=sm2[top-]+(ll)lj2*Mx(,ht[i]-lm+);
    }
      printf("%lld\n",ans);
    }
  return ;
}