POJ 1743 Musical Theme（后缀数组）

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

解题思路：

后缀数组的论文：https://wenku.baidu.com/view/5b886b1ea76e58fafab00374.html

大牛的论文18页给出了该类题型的解决办法。

就是二分答案P，然后我们找到一块块每个height值都大于等于P的区间[l,r]，这个区间内的任意两个后缀的LCP都>=P

我们找[l-1,r]的sa[]最小的的最大的差值，如果差值大于等于P，那么P可行，否则P不可行，进行二分就行。

 

本题由于题目只要找增减趋势相同的两段串，我们就储存每个点到下个点的增减的值，比如：

原串：1 2 3 4 5 6 7 8 9 10

变成：1 1 1 1 1 1 1 1 1 0

我们判断当前[l-1,r] 是否sa[]最大最小差值大于（没有等于）P即可。

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<map>
#include<set>

using namespace std;

#define ll long long
#define for1(i,a,b) for (int i=a;i<=b;i++)
#define for0(i,a,b) for (int i=a;i<b;i++)
#define pb push_back
#define fi first
#define se second
#define debug(x) printf("----Line %s----\n",#x)
#define pt(x,y) printf("%s = %d\n",#x,y)
#define INF 0x3f3f3f3f

const int N = 2e4+5;

int ss[N],wa[N],wb[N],rnk[N],height[N],sa[N],cnt[N];

void buildsa(int n,int m)
{
    int i,j,*x = wa,*y = wb,p;
    for0(i,0,m) cnt[i] = 0;
    for0(i,0,n) cnt[x[i]=ss[i]]++;
    for0(i,1,m) cnt[i] += cnt[i-1];
    for (int i=n-1;i>=0;i--) sa[--cnt[x[i]]] = i;

    for (j=1,p=1;p<n;j<<=1,m=p){
        for (i=n-j,p=0;i<n;i++) y[p++] = i;
        for0(i,0,n) if (sa[i]>=j) y[p++] = sa[i]-j;

        for0(i,0,m) cnt[i] = 0;
        for0(i,0,n) cnt[x[y[i]]]++;
        for0(i,1,m) cnt[i] += cnt[i-1];
        for (int i=n-1;i>=0;i--) sa[--cnt[x[y[i]]]] = y[i];

        swap(x,y);
        x[sa[0]] = 0;
        p = 1;
        for (i=1;i<n;i++)
            x[sa[i]] = (y[sa[i]]==y[sa[i-1]] && y[sa[i]+j]==y[sa[i-1]+j])? p-1:p++;
    }
}

void getheight(int n)
{
    int k = 0;
    for0(i,0,n) rnk[sa[i]] = i;
    for0(i,0,n){
        if (rnk[i]==0){height[rnk[i]] = k = 0;continue;  }
        if (k) k--;
        int j = sa[rnk[i]-1];
        while (i+k<n && j+k<n && ss[i+k]==ss[j+k]) k++;
        height[rnk[i]] = k;
    }
}

bool check(int k,int n)
{
    int l,r;
    for (int i=1;i<n;i++){
        if (height[i]>=k){
            int posmin=min(sa[i-1],sa[i]),posmax=max(sa[i-1],sa[i]);///最小的下标，最大的下标
            int j;
            for (j = i;j<n && height[j]>=k;j++){posmin = min(posmin,sa[j]);posmax = max(posmax,sa[j]);}
            if (posmax-posmin>k) return true;
            i = j;
        }
    }
    return false;
}

int main()
{
    int n;
    while (~scanf("%d",&n),n){
        for0(i,0,n) scanf("%d",ss+i);

        for0(i,0,n-1) ss[i] = ss[i+1]-ss[i] + (88*2+5);///后面181不加也行，只要让ss[n-1]最小就行
        ss[n-1] = 0;

        buildsa(n,300);///有效0~n-2,加入n-1防止越界，
        getheight(n);///求height 0~n-1

        int l = 0,r = n-1;
        while (l<=r){
            int k = l+r>>1;
            bool f = check(k,n);
            if (f) l = k+1;
            else r = k-1;
        }
        if (r+1>=5) printf("%d\n",r+1);
        else printf("0\n");
    }
    return 0;
}