POJ 1743Musical Theme（后缀数组+不可重叠最长子串）

/*
题意：
* POJ 1743 Musical Theme
* 有N(1 <= N <=20000)个音符的序列来表示一首乐曲，每个音符都是1..88范围内的整数，现在要找一个重复的主题。
* “主题”是整个音符序列的一个子串，它需要满足如下条件：
* 1.长度至少为5个音符
* 2.在乐曲中重复出现(可能经过转调，“转调”的意思是主题序列中每个音符都被加上或减去了同一个整数值。)
* 3.重复出现的同一主题不能有公共部分。
*
* 先转化成相邻两项的差值，然后就是找不可重叠重复子串。
* 做法就是二分答案LEN
* 然后根据height值进行分组
*/

思路：2009国家集训队论文上讲的很清楚，把sa数组按lcp进行分组，然后二分答案

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
const int maxn=200045;

int n,k;
int rank[maxn],tmp[maxn],lcp[maxn],sa[maxn];
int cmp(int i,int j){
    if(rank[i]!=rank[j])return rank[i]<rank[j];
    int ri=i+k<=n?rank[i+k]:-1;
    int rj=j+k<=n?rank[j+k]:-1;
    return ri<rj;
}
void getSa(int*s){
    for(int i=0;i<=n;i++){
        sa[i]=i;
        rank[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k<<=1){
        sort(sa,sa+n+1,cmp);
        tmp[sa[0]]=0;
        for(int i=1;i<=n;i++){
            tmp[sa[i]]=tmp[sa[i-1]]+cmp(sa[i-1],sa[i]);
        }
        for(int i=0;i<=n;i++){
            rank[i]=tmp[i];
        }
    }
}
void getLcp(int *s){
    for(int i=0;i<=n;i++)rank[sa[i]]=i;
    int h=0;
    lcp[0]=0;
    for(int i=0;i<n;i++){
        int j=sa[rank[i]-1];
        if(h)h--;
        while(i+h<n&&j+h<n&&s[i+h]==s[j+h])h++;
        lcp[rank[i]-1]=h;
    }
}
int s[maxn];
bool ok(int k){
    int mi=sa[0],mx=sa[0];
    for(int i=1;i<=n;i++){
        if(lcp[i-1]<k)mi=mx=sa[i];//按照lcp进行分组，lcp[i-1]!!!!
        else{
            mi=min(mi,sa[i]);
            mx=max(mx,sa[i]);
            if(mx-mi>k)return true;
        }
    }
    return false;
}
int main(){

    while(~scanf("%d",&n)&&n){
        for(int i=0;i<n;i++){
            scanf("%d",&s[i]);
        }
        for(int i=n-1;i>0;i--){
            s[i]=s[i]-s[i-1]+90;
        }
        n--;
        for(int i=0;i<n;i++){
            s[i]=s[i+1];
        }

        getSa(s);
        getLcp(s);

        int ans=-1;
        int l=0,r=n/2+1;
        while(l<=r){
            int mid=(l+r)/2;
            if(ok(mid)){
                ans=mid;
                l=mid+1;
            }
            else{
                r=mid-1;
            }
        }
        if(ans<4)printf("0\n");
        else printf("%d\n",ans+1);
    }
    return 0;
}
/*
15
1 1 1 1 1 1 2 2 2 2 2 2 2 2 2
6
*/