【后缀数组 不重叠重复最长子串】POJ - 1743 Musical Theme

Problem Description

题目大概是给n个数组成的串，有多个“重复”且不重叠的子串的长度大于等于5，两个子串重复当且仅当长度相等且每一位的数字差都相等，的最长子串
例如给你1 2 4 5 11 6 7 9 10 16。 1 2 4 5 11, 6 7 9 10 16这两个就是重复子串，它们的差1 2 1 6 -5 1 2 1 6.

思路：

参考文献：09年集训队论文后缀数组–处理字符串的有力工具
我们就把相邻数的差求出来存数组里，求这个数组重复最长子串(长度至少大于 4)，子串至少相隔一个位置。
二分答案枚举长度 k，按字典序从小到大的顺序 height[] 大于 k 的连续的分成一组，我们只需要判断每一组最左边后缀 和 最右边后缀 之间间距是否大于 k。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 200010;
const int inf = 0x3f3f3f3f;
int cntA[maxn], cntB[maxn], sa[maxn], tsa[maxn], A[maxn], B[maxn], height[maxn];
int Rank[maxn];
int n;
int a[maxn], ch[maxn];
void solve()//求sa[],Rank[],height[]
{
    for(int i = 0; i < 256; i++) cntA[i] = 0;
    for(int i = 1; i <= n; i++) cntA[ch[i-1]]++;
    for(int i = 1; i < 256; 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];
        //第二类tsa[i]第i小是哪一位。A[i]第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;
    }
}
bool Isok(int mid)
{
    int ma = -inf, mi = inf;
    for(int i = 2; i <= n; i++)
    {
        if(height[i] >= mid)//同一组公共前缀大于mid，找同一组最左边 和 最右边的子串判断它们是否距离差1
        {
            mi = min(mi, min(sa[i], sa[i-1]));
            ma = max(ma, max(sa[i], sa[i-1]));
            if(ma - mi > mid) return 1;//满足条件，return 1
        }else{//换新的一组，重新初始化
            ma = -inf, mi = inf;
        }
    }
    return 0;
}
int main()
{
    int i;
    while(~scanf("%d", &n))
    {
        if(!n) break;
        for(i = 0; i < n; i++)
        {
            scanf("%d", &a[i]);
        }
        for(i = 0; i < n; i++)//求相邻元素的差，不要负数所以+88
            ch[i] = a[i+1] - a[i] + 88;
        solve();//求sa[],Rank[],height[]
        int l = 4, r = n, ans = 0;
        while(l <= r)//二分答案
        {
            int mid = (l+r)/2;
            if(Isok(mid))//满足就增加长度
            ans = mid, l = mid+1;
            else r = mid-1;
        }
        if(!ans) printf("0\n");
        else
        printf("%d\n", ans+1);
    }
    return 0;
}