bzoj3998:[TJOI2015]弦论

Posted 2021-02-11 lcxer

传送门

T=0，就是求本质不同，后缀数组经典用法
T=1，二分解决
代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=5e5+10;
int n,k,num,x[maxn],a[maxn],y[maxn],sa[maxn],rk[maxn],h[maxn],opt,m='z',sum[maxn];char p[maxn];
int main()
{
    scanf("%s",p+1);read(opt),read(k);n=strlen(p+1);
    for(rg int i=1;i<=n;i++)a[x[i]=p[i]]++;
    for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
    for(rg int i=n;i;i--)sa[a[x[i]]--]=i;
    for(rg int k=1;k<=n;k<<=1,num=0)
    {
        for(rg int i=n-k+1;i<=n;i++)y[++num]=i;
        for(rg int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
        for(rg int i=1;i<=m;i++)a[i]=0;
        for(rg int i=1;i<=n;i++)a[x[i]]++;
        for(rg int i=1;i<=m;i++)a[i]+=a[i-1];
        for(rg int i=n;i;i--)sa[a[x[y[i]]]--]=y[i];
        for(rg int i=1;i<=n;i++)y[i]=x[i];
        num=x[sa[1]]=1;
        for(rg int i=2;i<=n;i++)
            if(y[sa[i]]!=y[sa[i-1]]||y[sa[i]+k]!=y[sa[i-1]+k])x[sa[i]]=++num;
            else x[sa[i]]=num;
        if(num>=n)break;m=num;
    }
    for(rg int i=1;i<=n;i++)rk[sa[i]]=i;
    for(rg int i=1,j,k=0;i<=n;h[rk[i++]]=k)
        for(k=k?k-1:k,j=sa[rk[i]-1];p[j+k]==p[i+k];k++);
    for(rg int i=1;i<=n;i++)sum[i]=sum[i-1]+n-sa[i]+1;
    if(!opt)
    {
        for(rg int i=1;i<=n;i++)
            if(n-sa[i]+1<k)k=k-(n-sa[i]+1)+h[i+1];
            else 
            {
                for(rg int j=sa[i];j<=sa[i]+k-1;j++)printf("%c",p[j]);printf("
");
                return 0;
            }
    }
    else
    {
        int L=1,R=n;
        for(int i=1;i<=n;i++)
        {
            int tmp=L;
            for(int j='a';j<='z';j++)
            {
                int l=tmp,r=R;
                while(l<=r)
                {
                    int mid=l+r>>1;
                    if(p[sa[mid]+i-1]>j)r=mid-1;
                    else l=mid+1;
                }
                long long t=sum[r]-sum[tmp-1]-1LL*(r-tmp+1)*(i-1);
                if(k<=r-tmp+1)
                {
                    for(int j=sa[tmp];j<=sa[tmp]+i-1;j++)printf("%c",p[j]);
                    return 0;
                }
                if(t>=k)
                {
                    L=tmp,R=r;
                    k-=r-tmp+1;
                    break;
                }
                tmp=r+1,k-=t;
            }
            if(n-sa[L]+1==i)L++;
        }
    }
    printf("-1
");
}