题解:POJ1743 Musical Theme 【后缀数组】

传送门

这个题目的题意就是差分一下,求差分数组的不可重叠的最长重复子串

这个题目的暴力很简单,就不多扯了(考试打了四个暴力的蒟蒻捂脸走开

于是我们考虑二分这个子串的长度,然后运用后缀数组进行判断

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

const int N=202000;
int n,m,sa[N],h[N],rk[N],s[N],a[N],x[N],y[N],c[N];

struct Sa {
    void qsort() {
        for(int i=1; i<=m; ++i)
            c[i]=0;
        for(int i=1; i<=n; ++i)
            ++c[x[i]];
        for(int i=1; i<=m; ++i)
            c[i]+=c[i-1];
        for(int i=n; i>=1; --i)
            sa[c[x[y[i]]]--]=y[i];
    }
    void get_sa() {
        for(int i=1; i<=n; ++i) 
            x[i]=s[i],y[i]=i;
        qsort(); 
        for(int k=1,num; num!=n; k<<=1,m=num) {
            num=0;
            for(int i=n-k+1; i<=n; ++i)
                y[++num]=i;
            for(int i=1; i<=n; ++i)
                if(sa[i]>k)
                    y[++num]=sa[i]-k;
            qsort();
            swap(x,y),num=0;
            for(int i=1; i<=n; ++i)
                x[sa[i]]=(y[sa[i]]==y[sa[i-1]])&&(y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
        }
        for(int i=1; i<=n; ++i)
            rk[sa[i]]=i;
    }
    void get_head() {
        int k=0;
        for(int i=1; i<=n; ++i) {
            if(rk[i]==1) continue;
            if(k) --k;
            int j=sa[rk[i]-1];
            while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])
                ++k;
            h[rk[i]]=k;
        }
    }
}SA;

int nmax,nmin,ans;

bool check(int x) {
    nmax=-10000;
    nmin=10000;
    for(int i=1; i<=n; i++) {
        if (h[i]>=x) {
            if(nmin==10000) {
                nmax=max(sa[i-1],nmax);
                nmin=min(sa[i-1],nmin);
            }
            nmax=max(sa[i],nmax);
            nmin=min(sa[i],nmin);
            if(nmax-nmin>x) return 1;
        } else nmax=-10000,nmin=10000;
    }
    return 0;
}

inline int read() {
    int s=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    return s;
}
int main() {

    n=read(),m=200;
    for(int i=1; i<=n; i++)
        a[i]=read();
    for(int i=1; i<=n; i++)
        s[i]=a[i]-a[i-1]+100;
    SA.get_sa(),SA.get_head();
    int l=1,r=n,mid;
    while(mid=l+r>>1,l^r)
        if(check(mid+1)) ans=l=mid+1;
        else r=mid;
    if(ans+1>=5)  cout<<ans+1;
    else cout<<0;

    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43464026/article/details/88351053