首页 > 代码库 > [poj 1743]后缀数组例题

[poj 1743]后缀数组例题

题目链接:http://poj.org/problem?id=1743

 

首先,musical theme只与前后位置的增减关系有关,而与绝对的数值无关,因此想到做一次差分。

然后对于差分后的数组,找到最长的出现两次(或两次以上)的一个子串即可。这个如果说两个子串可以交叉的话就好做了,直接取height的最大值即可,但是题目要求不能交叉,前几天一位师兄讲课刚讲了,可以用二分做。对于一个指定长度L,判断它是否可行,就用这个L去划分height数组,对于每个部分的分别看看最远的两个是否没有交叉就可以了。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxn=20005;

#define F(x) ((x)/3+((x)%3==1?0:tb))
#define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
int wa[maxn*3],wb[maxn*3],wv[maxn*3],wss[maxn*3];
int c0(int *r,int a,int b)
{
    return r[a]==r[b] && r[a+1]==r[b+1] && r[a+2]==r[b+2];
}
int c12(int k,int *r,int a,int b)
{
    if (k==2) return r[a]<r[b] || (r[a]==r[b]&&c12(1,r,a+1,b+1));
    else return r[a]<r[b] || (r[a]==r[b]&&wv[a+1]<wv[b+1]);
}
void sort(int *r,int *a,int *b,int n,int m)
{
    int i;
    for (i=0;i<n;i++) wv[i]=r[a[i]];
    for (i=0;i<m;i++) wss[i]=0;
    for (i=0;i<n;i++) wss[wv[i]]++;
    for (i=1;i<m;i++) wss[i]+=wss[i-1];
    for (i=n-1;i>=0;i--) b[--wss[wv[i]]]=a[i];
}
void dc3(int *r,int *sa,int n,int m)
{
    int i,j,*rn=r+n;
    int *san=sa+n,ta=0,tb=(n+1)/3,tbc=0,p;
    r[n]=r[n+1]=0;
    for (i=0;i<n;i++) if (i%3!=0) wa[tbc++]=i;
    sort(r+2,wa,wb,tbc,m);
    sort(r+1,wb,wa,tbc,m);
    sort(r,wa,wb,tbc,m);
    for (p=1,rn[F(wb[0])]=0,i=1;i<tbc;i++)
        rn[F(wb[i])]=c0(r,wb[i-1],wb[i])?p-1:p++;
    if (p<tbc) dc3(rn,san,tbc,p);
    else for (i=0;i<tbc;i++) san[rn[i]]=i;
    for (i=0;i<tbc;i++) if (san[i]<tb) wb[ta++]=san[i]*3;
    if (n%3==1) wb[ta++]=n-1;
    sort(r,wb,wa,ta,m);
    for (i=0;i<tbc;i++) wv[wb[i]=G(san[i])]=i;
    for (i=0,j=0,p=0;i<ta&&j<tbc;p++)
        sa[p]=c12(wb[j]%3,r,wa[i],wb[j])?wa[i++]:wb[j++];
    for (;i<ta;p++) sa[p]=wa[i++];
    for (;j<tbc;p++) sa[p]=wb[j++];
}
void da(int str[],int sa[],int rank[],int height[],int n,int m)
{
    for (int i=n;i<n*3;i++)
        str[i]=0;
    dc3(str,sa,n+1,m);
    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 (str[i+k]==str[j+k]) k++;
        height[rank[i]]=k;
    }
}
int a[maxn*3];
int ra[maxn*3],height[maxn*3],sa[maxn*3];
bool xk[maxn];
const int INF=0x3f3f3f3f;

bool check(int k,int n)
{
    for (int i=2;i<=n-1;i++)
        if (height[i]<k) xk[i]=false;
        else xk[i]=true;
    xk[n]=false;
    int left=INF,right=0;
    for (int i=2;i<=n;i++)
    {
        if (!xk[i])
        {
            if (right-left>k) return true;
            left=INF;
            right=0;
        }
        else
        {
            int zb1=sa[i-1];
            int zb2=sa[i];
            if (zb1<left) left=zb1;
            if (zb2<left) left=zb2;
            if (zb1>right) right=zb1;
            if (zb2>right) right=zb2;
        }
    }
    return false;
}

int solve(int n)
{
    // a[0..n-2]
    // ra[0..n-2]
    // sa[1..n-1]
    // height[2..n-1]
    int l=0,r=n-1;
    while (l<r)
    {
        int mid=(l+r+1)/2;
        if (check(mid,n)) l=mid;
        else r=mid-1;
    }
//    return l;
    if (l<4) return 0;
    return l+1;
}

int main()
{
    int n;
    while (~scanf("%d",&n) && n)
    {
        for (int i=0;i<n;i++) scanf("%d",&a[i]);
        for (int i=0;i<n-1;i++) a[i]=a[i+1]-a[i]+88;
//        for (int i=0;i<n-1;i++) printf("%d ",a[i]);
//        printf("\n");
        da(a,sa,ra,height,n-1,177);
        printf("%d\n",solve(n));
    }
    return 0;
}

 

[poj 1743]后缀数组例题