HDU 5008 Boring String Problem ( 后缀数组求本质不同第k大子串)

最新推荐文章于 2025-07-09 11:14:15 发布

原创最新推荐文章于 2025-07-09 11:14:15 发布 · 1.4k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

ACM补题

本文介绍了如何使用后缀数组解决BoringStringProblem，通过预处理sum数组记录不同字符串的个数，然后利用二分查找找到第k小子串。算法涉及到基数排序、高度计算等概念，适用于字符串搜索和子串计数问题。

Boring String Problem

Zeronera题解
预处理sum数组记录不同字符串的个数，即sum[i] = n- sa[i] + 1 -height[i] + sum[i-1] (n为原串长度)

对于每个k 若k > sum[n] 则输出0 0 ，即k大于不同子串的总数

否则，二分sum数组找到第k小子串所在的sa数组，即找到相应子串所位于的后缀。
l=sa[pos],r=sa[pos]+v+height[pos]-1;

但是该串的开始位置不一定是最小的，所以顺着sa数组要往后找是否还有更小的答案

至于为什么只要往后，因为当前找到子串的一定是该串在sa数组中第一次，之后若还出现，height数组会将其覆盖，所以要往后找

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
// sa[i]: 排名是i位的是第几个后缀
// rk[i]: 第i个后缀的排名是多少
// height[i]: sa[i]与sa[i-1]
const int N=100010;
char s[N];
int rk[N],sa[N],cnt[N],height[N];
int x[N],y[N];
int n,m;
void rsort()// x[i] 第一关键字 y[i] 第二关键字 基数排序
{
    for(int i=1;i<=m;i++) cnt[i]=0;
    for(int i=1;i<=n;i++) cnt[x[i]]++;
    for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for(int i=n;i;i--) sa[cnt[x[y[i]]]--]=y[i];
}
void SA()
{
    n=strlen(s+1);
    m=300;
    for(int i=1;i<=n;i++) x[i]=s[i],y[i]=i;
    rsort();
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k+1;i<=n;i++) y[++p]=i;// 第二关键字为空字符排在最前面
        for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
        rsort();swap(x,y);
        
        x[sa[1]]=1,p=1;
        for(int i=2;i<=n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p);
        if(p==n) break;
        m=p;
    }
    for(int i=1;i<=n;i++) rk[sa[i]]=i;
    // 求height
    height[1]=0;
    for(int i=1,j=0;i<=n;i++)
    {
        if(j) --j;
        while(s[i+j]==s[sa[rk[i]-1]+j]) j++;
        height[rk[i]]=j;
    }
}
ll sum[N];int q;
int main()
{
    while(~scanf("%s%d",s+1,&q))
    {
        SA();
        for(int i=1;i<=n;i++) sum[i]=n-sa[i]+1-height[i]+sum[i-1];
        
        
        ll l=0,r=0,v=0;
        while(q--)
        {
            scanf("%lld",&v);
            v=(l^r^v)+1;
            if(v>sum[n]) l=r=0;
            else
            {
                int pos=lower_bound(sum+1,sum+1+n,v)-sum;
                v-=sum[pos-1];
                l=sa[pos],r=sa[pos]+v+height[pos]-1;
                int len=r-l+1;
                pos++;
                while(height[pos]>=len)
                {
                    if(l>sa[pos])
                    {
                        l=sa[pos];
                        r=l+len-1;
                    }
                    pos++;
                }
            }
            printf("%lld %lld\n",l,r);
        }
        
    }
    return 0;
}