hdu 4416 后缀数组

最新推荐文章于 2014-10-14 18:55:17 发布

原创最新推荐文章于 2014-10-14 18:55:17 发布 · 406 阅读

0 ·

CC 4.0 BY-SA版权

字符串专栏收录该内容

17 篇文章

订阅专栏

本文详细介绍了一个高效的字符串匹配算法实现过程，包括核心数据结构和关键步骤的解析。通过具体代码示例，展示了如何进行字符串预处理及后处理计算，特别关注于如何通过优化减少不必要的比较操作。

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

首先参考这个链接的思路，讲的很清晰：http://blog.youkuaiyun.com/dgq8211/article/details/9850577

下面就是我写的代码了：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXD 333000
using namespace std;
int r[MAXD], Rank[MAXD], height[MAXD];
int sa[MAXD], wa[MAXD], wb[MAXD], WS[MAXD], wv[MAXD];
int n,len,lenA,casei;
__int64 ans;
char str[MAXD];
int cmp(int *p, int x, int y, int l)
{
    return p[x] == p[y] && p[x + l] == p[y + l];
}
void da(int n, int m)      //n为字符串长度，m为字符的最大值
{
    int i, j, p, *x = wa, *y = wb, *t;
    for(i = 0; i < m; i ++)
        WS[i] = 0;
    for(i = 0; i < n; i ++)
        ++ WS[x[i] = r[i]];
    for(i = 1; i < m; i ++)
        WS[i] += WS[i - 1];
    for(i = n - 1; i >= 0; i --)
        sa[-- WS[x[i]]] = i;
    for(p = 1, j = 1; p < n; j *= 2, m = p)
    {
        for(p = 0, 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 < n; i ++)
            wv[i] = x[y[i]];
        for(i = 0; i < m; i ++)
            WS[i] = 0;
        for(i = 0; i < n; i ++)
            ++ WS[wv[i]];
        for(i = 1; i < m; i ++)
            WS[i] += WS[i - 1];
        for(i = n - 1; i >= 0; i --)
            sa[-- WS[wv[i]]] = y[i];
        for(t = x, x = y, y = t, x[sa[0]] = 0, p = 1, i = 1; i < n; i ++)
            x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1: p ++;
    }
}
void calheight(int n)  //n为串的长度减1
{
    int i, j, k = 0;
    for(i = 1; i <= n; i ++)
        Rank[sa[i]] = i;
    for(i = 0; i < n; height[Rank[i ++]] = k)
        for(k ? -- k : 0, j = sa[Rank[i] - 1]; r[i + k] == r[j + k]; k ++);
}
int main()
{
    int i,j;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        casei++;
        scanf("%d",&n);
        ans=len=0;
        int num=100;
        for(i=0;i<=n;i++)
        {
            scanf("%s",str);
            int temp=strlen(str);
            if(i==0) lenA=temp;
            for(j=0;j<temp;j++)
            {
                r[len++]=str[j]-'a'+1;
            }
            r[len++]=num++;
        }
        r[len-1]=0;
        da(len,num);
        calheight(len-1);
        int lcpback=0;
        for(i=len-1;i>=0;i--)
        {
            if(sa[i]>lenA)
            {
                lcpback=height[i];
            }
            else
            {
                ans+=lenA-sa[i]-max(height[i],lcpback);
                lcpback=min(lcpback,height[i]);
            }
        }
        printf("Case %d: %I64d\n",casei,ans);
    }
    return 0;
}