4310: 跳蚤 后缀自动机+二分+hash

Description
很久很久以前，森林里住着一群跳蚤。一天，跳蚤国王得到了一个神秘的字符串，它想进行研究。首先，他会把串分成不超过 k 个子串，然后对于每个子串 S，他会从S的所有子串中选择字典序最大的那一个，并在选出来的 k 个子串中选择字典序最大的那一个。他称其为“魔力串”。现在他想找一个最优的分法让“魔力串”字典序最大。

题解：

首先可以用sam求出给定的串有多少个本质不同的子串，然后我们就可以二分字典序第 k k <script type="math/tex" id="MathJax-Element-564">k</script>小的子串，check的话从前往后不太好搞，若从后往前加字符的话那么每次只会增加一个后缀，对于每次新增加的后缀，我们可以用hash来判断它与二分出的串的字典序大小，来判定是否要断。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=200010;
const int inf=2147483647;
const unsigned int base=233;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
unsigned int Pow[100010],Hash[100010],_Hash[100010];
int k,len;
char s[Maxn];
int par[Maxn],mx[Maxn],son[Maxn][26];LL sum[Maxn],v[Maxn];
int tot=1,last=1;
LL cnt=0;
void extend(int x)
{
    int p=last,np=++tot;mx[np]=mx[p]+1;
    while(p&&!son[p][x])son[p][x]=np,p=par[p];
    if(!p)par[np]=1;
    else
    {
        int q=son[p][x];
        if(mx[p]+1==mx[q])par[np]=q;
        else
        {
            int nq=++tot;mx[nq]=mx[p]+1;
            for(int i=0;i<26;i++)son[nq][i]=son[q][i];
            par[nq]=par[q];
            par[q]=par[np]=nq;
            while(son[p][x]==q)son[p][x]=nq,p=par[p];
        }
    }
    last=np;
    cnt+=(LL)mx[np]-mx[par[np]];
}
int t[Maxn],b[Maxn],mxc=0;
char tmp[Maxn],ans[Maxn];int lt,lnow,lans;
bool cmp(int x)
{
    int l=1,r=min(lt,lnow);
    while(l<=r)
    {
        int mid=l+r>>1;
        if(Hash[x]-Hash[x+mid]*Pow[mid]==_Hash[1]-_Hash[1+mid]*Pow[mid])l=mid+1;
        else r=mid-1;
    }
    l--;
    if(l==min(lt,lnow))
    {
        if(lnow<=lt)return false;
        return true;
    }
    return s[x+l]>tmp[l+1];
}
void get_kth(int x,LL k)//第k小 
{
    if(k<=v[x])return;
    if(x!=1)k-=v[x];
    for(int i=0;i<26;i++)
    {
        int y=son[x][i];
        if(!y)continue;
        if(k<=sum[y])
        {
            tmp[++lt]=i+'a';
            get_kth(y,k);
            break;
        }
        else k-=sum[y];
    }
}
bool check(LL x)
{
    lt=0;
    get_kth(1,x);
    if(tmp[1]!=mxc+'a')return false;
    _Hash[lt+1]=0;
    for(int i=lt;i;i--)_Hash[i]=_Hash[i+1]*base+tmp[i]-'a';
    Hash[len+1]=0;
    int o=0;lnow=0;
    for(int i=len;i;i--)
    {
        Hash[i]=Hash[i+1]*base+s[i]-'a';lnow++;
        if(cmp(i))
        {
            o++;
            if(o==k)return false;
            Hash[i]=s[i]-'a';lnow=1;
        }
    }
    lans=lt;
    for(int i=1;i<=lans;i++)ans[i]=tmp[i];
    return true;
}
int main()
{
    k=read();
    scanf("%s",s+1);
    len=strlen(s+1);
    for(int i=1;i<=len;i++)extend(s[i]-'a'),mxc=max(mxc,s[i]-'a');
    for(int i=2;i<=tot;i++)sum[i]=v[i]=1;
    for(int i=2;i<=tot;i++)t[mx[i]]++;
    for(int i=1;i<=len;i++)t[i]+=t[i-1];
    for(int i=2;i<=tot;i++)b[t[mx[i]]--]=i;
    for(int i=tot-1;i;i--)
    {
        int x=b[i];
        for(int j=0;j<26;j++)
        {
            int y=son[x][j];
            if(!y)continue;
            sum[x]+=sum[y];
        }
    }
    Pow[0]=1;
    for(int i=1;i<=len;i++)Pow[i]=Pow[i-1]*base;
    LL l=1,r=cnt;
    while(l<=r)
    {
        LL mid=l+r>>1;
        if(check(mid))r=mid-1;
        else l=mid+1;
    }
    for(int i=1;i<=lans;i++)printf("%c",ans[i]);
}