POJ 1743 后缀数组(男人八题七)

来源:互联网 发布:网络电话回拨软件 编辑:程序博客网 时间:2024/05/22 10:47

题意

有一串音符,输出最长的两个不重叠的相似串的长度(如果小于5,则输出0)。相似的意思是两个串所有字符之差必须一样。

题解

很有意思的一道题。据说可以用HASH做,但是HASH看了半天都没看懂,只能用后缀数组了。
刚看到相似的时候我吓了一跳,不过如果转化成字符之差的话就舒服多了。我们可以用后一个字符减去前一个字符,这样可以得到一个差序列。如果两个差序列相等,则可以认为两个串相似。
对于这种找最长字符串的题,一般的套路就是二分。这里我们也二分一下长度就可以了。有一点需要注意的是,mx-mn>k,因为处理的是字符之差,如果等于的话,那么两个字符串的首尾便会重叠,所以不能取到=。

代码

#include <iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<vector>#include<cmath>#include<queue>#include<string>#include<set>#include<map>#include<bitset>#include<stack>#include<string>#define UP(i,l,h) for(int i=l;i<h;i++)#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)#define W(a) while(a)#define MEM(a,b) memset(a,b,sizeof(a))#define LL long long#define INF 0x3f3f3f3f3f3f3f3f#define MAXN 20050#define MOD 1000000007#define EPS 1e-3using namespace std;int a[MAXN];int sa[MAXN],t[MAXN],t2[MAXN],c[MAXN],height[MAXN],xrank[MAXN],n;void build_sa(int m) {    int i,*x=t,*y=t2;    UP(i,0,m) c[i]=0;    UP(i,0,n) c[x[i]=a[i]]++;    UP(i,1,m) c[i]+=c[i-1];    DOWN(i,n,0) sa[--c[x[i]]]=i;    for(int k=1; k<=n; k<<=1) {        int p=0;        UP(i,n-k,n) y[p++]=i;        UP(i,0,n) if(sa[i]>=k) y[p++]=sa[i]-k;        UP(i,0,m) c[i]=0;        UP(i,0,n) c[x[y[i]]]++;        UP(i,1,m) c[i]+=c[i-1];        DOWN(i,n,0) sa[--c[x[y[i]]]]=y[i];        swap(x,y);        p=1;        x[sa[0]]=0;        UP(i,1,n) x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;        if(p>=n) break;        m=p;    }}void getHeight() {    int i,j,k=0;    UP(i,0,n) xrank[sa[i]]=i;    UP(i,0,n-1) {        if(k) k--;        int j=sa[xrank[i]-1];        W(a[i+k]==a[j+k]) k++;        height[xrank[i]]=k;//        cout<<i<<" "<<k<<endl;    }}bool good(int k) {    int mx=0,mi=INF;//    cout<<k<<endl;    UP(i,1,n) {        if(height[i]>=k) {            mi=min(mi,min(sa[i],sa[i-1]));            mx=max(mx,max(sa[i],sa[i-1]));            if(mx-mi>k) return true;        }    }    return false;}int main() {    W(~scanf("%d",&n)) {        if(n==0) break;        UP(i,0,n) scanf("%d",&a[i]);        UP(i,0,n-1) {            a[i]=a[i+1]-a[i]+88;//            cout<<a[i]<<endl;        }        a[n]=0;        build_sa(200);        getHeight();        int l=4,h=n-1,ans=-1;        W(l<=h) {            int m=(l+h)/2;            if(good(m)) {                l=m+1;                ans=m;            } else {                h=m-1;            }        }        printf("%d\n",ans+1);    }}