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

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

先用后缀数组处理出 ht[i]；

用单调栈维护当前位置 ht[i] 对之前的 ht[j] 取 min 的结果，也就是当前的后缀与之前后缀的LCP，其中长度 >= K 的加到答案；

因为单调栈中是一段一段阶梯状的，只存了一段端点的位置，所以再记录一个 cnt 表示这一段的长度，算贡献时乘上 cnt；

因为是两个串之间，所以先统计 B 在 A 排名前的答案，再重复一遍统计 A 在 B 排名前的答案；

但是 ht[i] 是 sa[i] 和 sa[i-1] 的LCP，所以 ht[i] 是否计入贡献应该考虑 i-1 位置...突然变得很麻烦，不太会弄了...

于是参考了一下TJ(囧)，原来就是判断一下 i-1 是否要被统计，如果要统计 B 而 i-1 是 B 中的，就把 ht[i] 也累加到 sum 中；

还有一个很好的操作是如果 ht[i] < K，那么取 min 显然都会取成 < K 的，没贡献了，所以直接 sum=0 , top = 0，就省去了 max(0,ht[i]-K+1) 的分类麻烦。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=1e5+,xxn=(xn<<);//xxn
int n,m,tax[xxn],sa[xxn],rk[xxn],tp[xxn],ht[xxn],sta[xxn],top,cnt[xxn];
char a[xn],b[xn],s[xxn];
void Rsort()
{
  for(int i=;i<=m;i++)tax[i]=;
  for(int i=;i<=n;i++)tax[rk[tp[i]]]++;
  for(int i=;i<=m;i++)tax[i]+=tax[i-];
  for(int i=n;i;i--)sa[tax[rk[tp[i]]]--]=tp[i];
}
void work()
{
  for(int i=;i<=n;i++)rk[i]=s[i],tp[i]=i;
  Rsort();
  for(int k=;k<=n;k<<=)
    {
      int num=;
      for(int i=n-k+;i<=n;i++)tp[++num]=i;
      for(int i=;i<=n;i++)
    if(sa[i]>k)tp[++num]=sa[i]-k;
      Rsort();
      memcpy(tp,rk,sizeof rk);//swap(rk,tp);
      rk[sa[]]=; num=;
      for(int i=;i<=n;i++)
    rk[sa[i]]=(tp[sa[i]]==tp[sa[i-]]&&tp[sa[i]+k]==tp[sa[i-]+k])?num:++num;
      if(num==n)break;
      m=num;
    }
}
void get()
{
  int k=; ht[]=;
  for(int i=;i<=n;i++)
    {
      if(rk[i]==)continue;
      if(k)k--; int j=sa[rk[i]-];
      while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])k++;
      ht[rk[i]]=k;
    }
}
int main()
{
  int K;
  while()
    {
      scanf("%d",&K); if(!K)return ;
      scanf("%s",a+); int l1=strlen(a+);
      scanf("%s",b+); int l2=strlen(b+);
      n=l1+l2+;
      for(int i=;i<=l1;i++)s[i]=a[i]; s[l1+]='z'+;
      for(int i=;i<=l2;i++)s[l1++i]=b[i];
      m=; work(); get();
      ll ans=;
      ll sum=; top=;
      for(int i=,y;i<=n;i++)
    {
      cnt[i]=;
      if(ht[i]<K){top=; sum=; continue;}//min<K
      while(ht[i]<ht[y=sta[top]]&&top)
        {
          sum-=(ll)cnt[y]*(ht[y]-K+);
          sum+=(ll)cnt[y]*(ht[i]-K+);//
          top--; cnt[i]+=cnt[y];
        }
      sta[++top]=i;
      if(sa[i-]>l1+)sum+=ht[i]-K+,cnt[i]++;//cal(i-1):ht[i]
      if(sa[i]<=l1)ans+=sum;
    }
      sum=; top=;
      for(int i=,y;i<=n;i++)
    {
      cnt[i]=;
      if(ht[i]<K){top=; sum=; continue;}//min<K
      while(ht[i]<ht[y=sta[top]]&&top)
        {
          sum-=(ll)cnt[y]*(ht[y]-K+);
          sum+=(ll)cnt[y]*(ht[i]-K+);
          top--; cnt[i]+=cnt[y];
        }
      sta[++top]=i;
      if(sa[i-]<=l1)sum+=ht[i]-K+,cnt[i]++;
      if(sa[i]>l1+)ans+=sum;
    }
      printf("%lld\n",ans);
    }
  return ;
}