SPOJ 705 New Distinct Substrings 后缀数组

最新推荐文章于 2019-08-31 11:55:10 发布

原创最新推荐文章于 2019-08-31 11:55:10 发布 · 312 阅读

0 ·

CC 4.0 BY-SA版权

----------- 算法 ---------- 同时被 3 个专栏收录

138 篇文章

订阅专栏

字符串

35 篇文章

订阅专栏

字符串 -------- 后缀数组

6 篇文章

订阅专栏

本文介绍了一个解决SPOJ SUBST1问题的方法，通过使用C++实现的字符串处理算法，包括构建后缀数组和计算高度数组来找出所有不同子串的数量。该算法适用于字符串相似性和模式匹配等问题。

题目链接：http://www.spoj.com/problems/SUBST1/

和SPOJ 694 一样

代码如下：

#include <cstdio>
#include <cstring>
#define sf scanf
#define pf printf
using namespace std;
const int maxn = 50000 + 5;

int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(int *r,int *sa,int n,int 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(j=1,p=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,p=1,x[sa[0]]=0,i=1;i<n;i++)
        x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
    return;
}
int rank[maxn],height[maxn];
void calheight(int *r,int *sa,int n)
{
    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++);
    return;
}

int sa[maxn],rr[maxn];
char s[maxn];
typedef long long LL;
int main(){
    int T;
    sf("%d",&T);
    while( T-- ){
        sf("%s",s);
        int len = strlen(s);
        for(int i = 0;i <= len;++i) rr[i] = s[i];
        da(rr,sa,len + 1,256);
        calheight(rr,sa,len);
        LL cnt = (LL)len * (len + 1) / 2;
        for(int i = 1;i <= len;++i) cnt -= height[i];
        pf("%lld\n",cnt);
    }
    return  0;
}