POJ-3415-Common Substrings(后缀数组+单调栈)

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


求两串中长度大于k的公共子串有多少个。


公共子串可以通过height求,中间分隔连接两串,将height[i]>=k进行分组,对于一组内的height[i],且sa[i]属于a串,需要找到ji]-k),采用单调栈维护一个栈顶最小的height[i],大于栈顶压入,小于更新。每次针对a/b串找前面的b/a串,跑两次。



//#include 
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define INF 0x3f3f3f3f
using namespace std;
const int MAXN = 4e5+7;
int t1[MAXN],t2[MAXN],c[MAXN];
int sa[MAXN],ra[MAXN],height[MAXN];
char str1[MAXN],str2[MAXN];
int len1,len2,len;
int num[MAXN];
int cnt[MAXN];
bool cmp(int *r, int a, int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void DA(int str[], int n, int m)
{
    n++;
    int i,j,p,*x=t1,*y=t2;
    for(i=0; i=0; --i)sa[ --c[ x[i] ] ]=i;
    for(j=1; j<=n; j<<=1)
    {
        p=0;
        for(i=n-j; i=j)y[p++]=sa[i]-j;
        for(i=0; i=0; --i)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        for(i=1; i=height[i])
                {
                    top--;
                    sum-=1LL*(que[top][0]-height[i])*que[top][1];
                    cnt+=que[top][1];
                }
                que[top][0]=height[i];
                que[top++][1]=cnt;
                if(sa[i]>len1)ans+=sum;
            }
        }
        top=0;
        for(int i=1; i<=len; ++i)
        {
            if(height[i]len1)cnt=1,sum+=height[i]-k+1;
                while(top!=0 && que[top-1][0]>=height[i])
                {
                    top--;
                    sum-=1LL*(que[top][0]-height[i])*que[top][1];
                    cnt+=que[top][1];
                }
                que[top][0]=height[i];
                que[top++][1]=cnt;
                if(sa[i]


你可能感兴趣的:(POJ,字符串)