[USACO5.1] Musical Themes

后缀数组求最长重复且不重叠子串。

poj 1743 传送门

洛谷 P2743 传送门

1.子串可以“变调”（即1 3 6和3 5 8视作相同）。解决办法：求字符串相邻元素的差形成新串。用新字符串求解最长重复子串即可。

2.最长重复子串不能重叠。解决办法：用sa数组判断开始位置。

倍增答案即可，从1到n枚举height，记录子串开始的最左端、最右端。

如果找到了两个后缀，其公共前缀长度大于k，且其开始位置之间的间隔大于k，就满足条件。

由height数组的性质可得：要使x、y的公共前缀长度大于k，则需要h[x+1]、h[x+2]......h[y]全部大于k。

所以只要有一个没有大于k，就得重新开始。即：重置子串开始的最左端、最右端。

最后答案需要+1，并判断是否大于等于5（题意要求）。

注意poj的数据范围比洛谷上大，而且有多组测试数据。

下面只给出poj1743的代码。

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

int n,ans;
int s[20005];
int sa[20005],rk[20005];
int tr[20005],h[20005];

int cmp(int x,int y,int k)
{
    if(x+k>n||y+k>n)return 0;
    return rk[x]==rk[y]&&rk[x+k]==rk[y+k];
}

void cal()
{
    int i,cnt;
    for(i=1;i<=n;i++)h[s[i]]++;
    for(cnt=0,i=1;i<=200;i++)if(h[i])tr[i]=++cnt;
    for(i=1;i<=200;i++)h[i]+=h[i-1];
    for(i=1;i<=n;i++)rk[i]=tr[s[i]],sa[h[s[i]]--]=i;
    for(int k=1;cnt!=n;k<<=1)
    {
        for(i=1;i<=n;i++)h[i]=0;
        for(i=1;i<=n;i++)h[rk[i]]++;
        for(i=1;i<=n;i++)h[i]+=h[i-1];
        for(i=n;i;i--)if(sa[i]>k)tr[sa[i]-k]=h[rk[sa[i]-k]]--;
        for(i=1;i<=k;i++)tr[n-i+1]=h[rk[n-i+1]]--;
        for(i=1;i<=n;i++)sa[tr[i]]=i;
        for(cnt=0,i=1;i<=n;i++)tr[sa[i]]=cmp(sa[i],sa[i-1],k)?cnt:++cnt;
        for(i=1;i<=n;i++)rk[i]=tr[i];
    }
    for(i=1;i<=n;i++)h[i]=0;
    for(i=1;i<=n;i++)
    {
        if(rk[i]==1)continue;
        for(int j=max(1,h[rk[i-1]]-1);;j++)
        {
            if(s[i+j-1]==s[sa[rk[i]-1]+j-1])h[rk[i]]=j;
            else break;
        }
    }
}

int check(int k)
{
    if(k>n)return 0;
    int l,r;
    l=r=sa[1];
    for(int i=2;i<=n;i++)
    {
        if(h[i]>=k)
        {
            l=min(l,sa[i]);
            r=max(r,sa[i]);
            if(r-l>k)return 1;
        }else l=r=sa[i];
    }
    return 0;
}

int main()
{
    scanf("%d",&n);
    while(n)
    {
        memset(h,0,sizeof(h));
        memset(tr,0,sizeof(tr));
        memset(rk,0,sizeof(rk));
        memset(sa,0,sizeof(sa));
        ans=0;
        for(int i=1;i<=n;i++)scanf("%d",&s[i]);
        for(int i=1;i<n;i++)s[i]=s[i+1]-s[i]+90;
        cal();
        for(int i=20;i>=0;i--)
            if(check(ans|(1<<i)))ans|=(1<<i);
        ans=(ans+1)>=5?(ans+1):0;
        printf("%d
",ans);
        scanf("%d",&n);
    }
    return 0;
}