Musical Theme POJ - 1743 (后缀数组求不可重叠最长重复子串）

题目 https://cn.vjudge.net/problem/POJ-1743 

题意

给定一个字符串，求最长重复子串，这两个子串不能重叠。
思路
这题比上一题稍复杂一点。先二分答案，把题目变成判定性问题：判断是否
存在两个长度为 k 的子串是相同的，且不重叠。解决这个问题的关键还是利用
height 数组。把排序后的后缀分成若干组，其中每组的后缀之间的 height 值都
不小于 k。例如，字符串为“aabaaaab”，当 k=2 时，后缀分成了 4 组，

容易看出，有希望成为最长公共前缀不小于 k 的两个后缀一定在同一组。然
后对于每组后缀，只须判断每个后缀的 sa 值的最大值和最小值之差是否不小于
k。如果有一组满足，则说明存在，否则不存在。整个做法的时间复杂度为
O(nlogn)。本题中利用 height 值对后缀进行分组的方法很常用，请读者认真体
会。

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>

using namespace std;

typedef long long ll;
const int maxn = 200010;
int cntA[maxn],cntB[maxn],sa[maxn],tsa[maxn],A[maxn],B[maxn],height[maxn];
int a[maxn];
int Rank[maxn];
ll n;
int ch[maxn];

void solve()
{
    for(int i = 0;i < 222;i++) cntA[i] = 0;
    for(int i = 1;i <= n;i++) cntA[ch[i-1]]++;
    for(int i = 1;i < 222;i++) cntA[i] += cntA[i-1];
    for(int i = n;i;i--) sa[cntA[ch[i-1]]--] = i;

    Rank[sa[1]] = 1;
    for(int i = 2;i <= n;i++)
    {
        Rank[sa[i]] = Rank[sa[i-1]];
        if(ch[sa[i]-1] != ch[sa[i-1]-1]) Rank[sa[i]]++;
    }

    for(int l = 1;Rank[sa[n]] < n;l <<= 1)
    {
        memset(cntA,0,sizeof(cntA));
        memset(cntB,0,sizeof(cntB));
        for(int i = 1;i <= n;i++)
        {
            cntA[A[i] = Rank[i]]++;
            cntB[B[i] = (i+l <= n)?Rank[i+l]:0]++;
        }
        for(int i = 1;i <= n;i++) cntB[i] += cntB[i-1];
        for(int i = n;i; i--) tsa[cntB[B[i]]--] = i;
        for(int i = 1;i <= n;i++) cntA[i] += cntA[i-1];
        for(int i = n; i;i--) sa[cntA[A[tsa[i]]]--] = tsa[i];

        Rank[sa[1]] = 1;
        for(int i = 2;i <= n;i++)
        {
            Rank[sa[i]] = Rank[sa[i-1]];
            if(A[sa[i]] != A[sa[i-1]] || B[sa[i]] != B[sa[i-1]]) Rank[sa[i]]++;
        }

    }
    for(int i = 1,j = 0;i <= n;i++)
    {
        if(j) j--;
        while(ch[i+j-1] == ch[sa[Rank[i]-1] + j - 1]) j++;
        height[Rank[i]] = j;
    }
}
int ok(int x)
{
    int ma,mi;
    for(int i = 1;i <= n;)
    {
        while(height[i] < x&&i <= n) i++;
        ma = mi = sa[i-1];
        while(height[i] >= x && i<= n)
        {
            ma = max(ma,sa[i]);
            mi = min(mi,sa[i]);
            i++;
        }
        if(ma - mi > x) return 1;
    }
    return 0;
}
int main()
{
    while(scanf("%d", &n) != EOF&&n)
    {
        scanf("%d",&a[0]);
        for(int i = 1;i < n;i++)
        {
            scanf("%d", &a[i]);
            ch[i-1] = a[i] - a[i-1];
            ch[i-1] += 100;
        }
        n--;
        solve();
        int l = 0,r = n+1;
        while(l+1 < r)
        {
            int mid = (l+r)/2;
            if(ok(mid))
            {
                l = mid;
            }
            else
            {
                r = mid;
            }
        }
        int qwe = 0;
        if(ok(r)) printf("%d\n",r+1>=4?r+1 : qwe);
        else printf("%d\n",l+1 >= 4?l+1 : qwe);
    }
}