HDU-5769-Substring（后缀数组）

最新推荐文章于 2024-05-09 00:45:37 发布

原创最新推荐文章于 2024-05-09 00:45:37 发布 · 332 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #c++ #HDU

HDU 同时被 2 个专栏收录

106 篇文章

订阅专栏

字符串

10 篇文章

订阅专栏

本文介绍了一种用于求解特定字符串问题的算法实现，该算法利用高级数据结构来高效地找出包含指定字符的所有不重复子串数量。通过详细的代码示例展示了如何使用数列排序和高度数组来优化匹配过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接：http://acm.hdu.edu.cn/showproblem.php?pid=5769

题意：给定字符串str ('a'-'z')和字符c，问str中包含c的不重复字串有多少。

题解：

CODE：

#include <bits/stdc++.h>
using namespace std;
#define bug cout<<"bug"<<endl
#define INF 0x3f3f3f3f
#define mod 1000000007
const int MAXN = 1e5+7;
int t1[MAXN],t2[MAXN],c[MAXN];
int sa[MAXN],ra[MAXN],height[MAXN];
char st[MAXN],s[MAXN];
int num[MAXN];
bool cmp(int *r, int a, int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void DA(int str[], int n, int m)
{
    n++;
    int i,j,p,*x=t1,*y=t2;
    for(i=0; i<m; ++i)c[i]=0;
    for(i=0; i<n; ++i)c[x[i]=str[i]]++;
    for(i=1; i<m; ++i)c[i]+=c[i-1];
    for(i=n-1; i>=0; --i)sa[ --c[ x[i] ] ]=i;
    for(j=1; j<=n; j<<=1)
    {
        p=0;
        for(i=n-j; i<n; ++i)y[p++]=i;
        for(i=0; i<n; ++i)if(sa[i]>=j)y[p++]=sa[i]-j;

        for(i=0; i<m; ++i)c[i]=0;
        for(i=0; i<n; ++i)c[x[y[i]]]++;
        for(i=1; i<m; ++i)c[i]+=c[i-1];
        for(i=n-1; i>=0; --i)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        for(i=1; i<n; ++i)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        m=p;
    }
    int k=0;
    n--;
    for(i=0; i<=n; ++i)ra[sa[i]]=i;
    for(i=0; i<n; ++i)
    {
        if(k)--k;
        j=sa[ra[i]-1];
        while(str[i+k]==str[j+k])k++;
        height[ra[i]]=k;
    }
}
int nxt[MAXN];
int main()
{
    int T,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s);
        scanf("%s",st);
        int len = strlen(st);
        for(int i=0; i<len; ++i)
            num[i]=st[i]-'a'+1;
        num[len]=0;
        DA(num,len,30);
        int poi=len;
        for(int i=len-1; i>=0; --i)
        {
            if(st[i]==s[0])poi=i;
            nxt[i]=poi;
        }
        long long ans=0;
        for(int i=1; i<=len; ++i)
            ans=ans+(len-max(nxt[sa[i]],sa[i]+height[i]));
        printf("Case #%d: %I64d\n",cas++,ans);
    }
    return 0;
}
/*
a
abaca
*/