POJ 3415 Common Substrings(后缀数组:后缀公共前缀个数)

POJ 3415 Common Substrings(后缀数组:后缀公共前缀个数)

http://poj.org/problem?id=3415

题意:

A substring of astring T is defined as:

T(i,k)=TiTi+1...Ti+k-1, 1≤i≤i+k-1≤|T|.

Given two stringsA, B and one integer K, we define S, a set of triples (i, j, k):

S = {(i, j, k) | k≥K, A(i, k)=B(j, k)}.

要我们求集合S中元素的个数.

分析: 罗穗骞《后缀数组——处理字符串的有力工具》论文例题.

其实就是求A的任一后缀和B的任一后缀能贡献的LCP>=K的个数和.

依然老做法,将A与B中间填个字符’$’,然后连接他们形成新字符串.然后求height.

       将height分组,然后对于每组内的height最小值必须>=k才行,那么这组内的任意一个A与B的后缀可以贡献的子串是LCP(ia,ib)-k+1个.(想想是不是).不过如果穷举每组的话可能会超时,所以这里用到了单调栈来处理,使得在O(n)时间内处理完.

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=200000+100;
struct SuffixArray
{
    char s[maxn];
    int sa[maxn],rank[maxn],height[maxn];
    int t1[maxn],t2[maxn],c[maxn],n;
    void build_sa(int m)
    {
        int i,*x=t1,*y=t2;
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[i]=s[i]]++;
        for(i=1;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
        for(int k=1;k<=n;k<<=1)
        {
            int p=0;
            for(i=n-k;i<n;i++) y[p++]=i;
            for(i=0;i<n;i++)if(sa[i]>=k) y[p++]=sa[i]-k;
            for(i=0;i<m;i++) c[i]=0;
            for(i=0;i<n;i++) c[x[y[i]]]++;
            for(i=1;i<m;i++) c[i]+=c[i-1];
            for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
            swap(x,y);
            p=1,x[sa[0]]=0;
            for(i=1;i<n;i++)
                x[sa[i]]= y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
            if(p>=n) break;
            m=p;
        }
    }
    void build_height()
    {
        int i,j,k=0;
        for(i=0;i<n;i++) rank[sa[i]]=i;
        for(i=0;i<n;i++)
        {
            if(k)k--;
            j=sa[rank[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[rank[i]]=k;
        }
    }
}sa;
int f[maxn];
int a[maxn],b[maxn],c;
long long ss,ans;

int main()
{
    int i,k,l,n,t;
    while(scanf("%d",&k)==1&&k)
    {
        scanf("%s",sa.s);
        l=strlen(sa.s);
        sa.s[l]=1;
        scanf("%s",sa.s+l+1);
        n=strlen(sa.s);
        sa.s[n]=0;
        sa.n=n+1;
        sa.build_sa(128);
        sa.build_height();

        for(i=2;i<=n;i++)
        {
            f[i]=sa.sa[i]<l;//f[i]=true 表示字典序第i个后缀是输入A串的
            sa.height[i]-=k-1;
            if(sa.height[i]<0) sa.height[i]=0;
        }
        sa.height[n+1]=0;
        a[0]=-1;ans=0;
        for(t=0;t<=1;t++)
        for(c=0,ss=0,i=2;i<=n;i++)//a这个栈从c=0到c=c是依height值增加的
        {
            if(f[i]!=t) ans+=ss;//t=0时遇到A串后缀,或t=1时遇到B串后缀,都要加上之前的结果
            c++;//c是栈的指针
            a[c]=sa.height[i+1];//a是用来保存当前扫描到的height值的栈,且a[c]是当前最小值
            b[c]=f[i]==t;//字典序第i个后缀属于A串且t==1或i后缀输入B且t==0,那么b[c]=1,否则b[c]=0
            ss+=(long long)a[c]*b[c];
            while(a[c-1]>=a[c])//维护单调性
            {
            ss-=(long long)(a[c-1]-a[c])*b[c-1];//减去刚刚多加的那部分
            a[c-1]=a[c];
            b[c-1]+=b[c];//这个的意思是一个值可能是多个区间的最小值
            c--;
            }
        }
        printf("%I64d\n",ans);
    }
}