POJ 3415 Common Substrings

一道很巧妙的后缀数组题。

按照后缀数组的一般思维，我们要求的是两个字符串中长度不小于k的子串长度，所以我们可以将两个字符串通过'$"拼接起来，这样问题就转化成了一个height数组中的关系。

首先，height数组一定要分块，分块之后，我们就要求块中任意一对后缀的公共前缀长度。

一开始自己想的比较简单，想用暴力的方法做过，就弄了一个ST表，每次用O(1)的时间查询一下lcp的值。但是当块比较大时，就变成O(n^2)的了，无情TLE。（这个测试用例也很好想，两个长度达到10w的x串）

这时候我们就要利用lcp的性质来搞了。

求lcp(suffix(i),suffix(j))，那么我们需要求height[rank[i]+1]到height[rank[j]]的最小值。

那么近乎O(n)的算法就是来维护一个单调栈。

比方说一块height值是2，3，4，5。2，3，4是A串，5是B串。那对应的lcp值就是2，3，4。直接累加就就可以搞了。

考虑一个常规的情况，B串前A串的height值为2，3，6，5，4。

对于2，3，6入栈，我们只需要将和累加就好了。

再加入5，那么我们的6就相当于多算了（应该算5）。

那栈就变成2，3，5（2），括号中的代表个数。

再加上4的话，就变成2，3，4（3）。

        大概就是这样了吧，跑两遍就可以了。。。

#include<cstdio>
#include<cstring>
using namespace std;
const int nMax = 200005;

int  num[nMax];
int sa[nMax], rank[nMax], height[nMax];
int wa[nMax], wb[nMax], wv[nMax], wd[nMax];

int cmp(int *r, int a, int b, int l){
    return r[a] == r[b] && r[a+l] == r[b+l];
}
int min(int a,int b){
    return a<b ? a:b;
}

void da(int *r, int n, int m){          //  倍增算法 r为待匹配数组  n为总长度 m为字符范围
    int i, j, p, *x = wa, *y = wb, *t;
    for(i = 0; i < m; i ++) wd[i] = 0;
    for(i = 0; i < n; i ++) wd[x[i]=r[i]] ++;
    for(i = 1; i < m; i ++) wd[i] += wd[i-1];
    for(i = n-1; i >= 0; i --) sa[-- wd[x[i]]] = i;
    for(j = 1, p = 1; p < n; j *= 2, m = p){
        for(p = 0, i = n-j; i < n; i ++) y[p ++] = i;
        for(i = 0; i < n; i ++) if(sa[i] >= j) y[p ++] = sa[i] - j;
        for(i = 0; i < n; i ++) wv[i] = x[y[i]];
        for(i = 0; i < m; i ++) wd[i] = 0;
        for(i = 0; i < n; i ++) wd[wv[i]] ++;
        for(i = 1; i < m; i ++) wd[i] += wd[i-1];
        for(i = n-1; i >= 0; i --) sa[-- wd[wv[i]]] = y[i];
        for(t = x, x = y, y = t, p = 1, x[sa[0]] = 0, i = 1; i < n; i ++){
            x[sa[i]] = cmp(y, sa[i-1], sa[i], j) ? p - 1: p ++;
        }
    }
}

void calHeight(int *r, int n){           //  求height数组。
    int i, j, k = 0;
    for(i = 1; i <= n; i ++) rank[sa[i]] = i;
    for(i = 0; i < n; height[rank[i ++]] = k){
        for(k ? k -- : 0, j = sa[rank[i]-1]; r[i+k] == r[j+k]; k ++);
    }
}

char s[nMax];
int stack[nMax][2];

int main(){
    int k;
    int len1,len2;
    long long ans;
    while(scanf("%d",&k)!=EOF&&k){
        memset(height,0,sizeof(height));
        ans=0;
        int top;
        long long sum;
        int cnt;
        scanf("%s",s);
        len1=(int)strlen(s);
        s[len1]='$';
        scanf("%s",s+len1+1);
        //puts(s);
        len2=(int)strlen(s);
        for(int i=0;i<len2;i++)  num[i]=s[i];
        num[len2]=0;
        da(num,len2+1,250);
        calHeight(num,len2);
        //printf("%d\n",len2);
        //for(int i=2;i<=len2;i++)  printf("%d ",height[i]);
        //printf("\n");
        top=sum=0;
        for(int i=1;i<=len2;i++){
            if(height[i]<k){
                top=sum=0;
            }else{
                cnt=0;
                if(sa[i-1]<len1)  cnt++,sum+=height[i]-k+1;
                while(top>0&&height[i]<=stack[top-1][0]){
                    top--;
                    sum-=stack[top][1]*(stack[top][0]-height[i]);
                    cnt+=stack[top][1];
                }
                stack[top][0]=height[i],stack[top++][1]=cnt;
                if(sa[i]>len1)  ans+=sum;
            }
        }
        //printf("%lld\n",ans);
        top=sum=0;
        for(int i=1;i<=len2;i++){
            if(height[i]<k){
                top=sum=0;
            }else{
                cnt=0;
                if(sa[i-1]>len1)  cnt++,sum+=height[i]-k+1;
                while(top>0&&height[i]<=stack[top-1][0]){
                    top--;
                    sum-=stack[top][1]*(stack[top][0]-height[i]);
                    cnt+=stack[top][1];
                }
                stack[top][0]=height[i],stack[top++][1]=cnt;
                if(sa[i]<len1)  ans+=sum;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}