关闭

spoj 694&&705 后缀数组

579人阅读 评论(0) 收藏 举报
每个子串一定是某个后缀的前缀,那么原问题等价于求所有后缀之间的不相同的前缀的个数。如果所有的后缀按照 suffix(sa[1]), suffix(sa[2]),suffix(sa[3]), …… ,suffix(sa[n])的顺序计算,不难发现,对于每一次新加进来的后缀 suffix(sa[k]),它将产生 n-sa[k]+1 个新的前缀。但是其中有height[k]个是和前面的字符串的前缀是相同的。所以 suffix(sa[k])将“贡献”出 n-sa[k]+1- height[k]个不同的子串。累加后便是原问题的答案。这个做法的时间复杂度为 O(n)

#include<iostream>
#include<cstring>
using namespace std;

const int maxn=1010;
int a[maxn],b[maxn],s[maxn],c[maxn];
int *rank,*height,*sa=s+1;

void sortandrank(int *a1,int *a2,int n,int &m,int j)
{
    int i;memset(c,0,sizeof(c));
    for(i=0;i<n;i++) c[a1[i]]++;
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[a1[a2[i]]]]=a2[i];
    a2[sa[0]]=m=0;
    for(i=1;i<n;i++) a2[sa[i]]=a1[sa[i-1]]==a1[sa[i]]&&a1[sa[i-1]+j]==a1[sa[i]+j]?m:++m;
}
void da(char*str,int n,int m)
{
    int *a1=a,*a2=b,*tmp;
    int i,j,p;
    for(i=0;i<n;i++) a1[i]=i,a2[i]=str[i];
    a1[n]=a2[n]=-1;
    sortandrank(a2,a1,n,m,0);
    for(j=1;m<n-1;j<<=1)
    {
        p=0;
        for(i=n-j;i<n;i++) a2[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) a2[p++]=sa[i]-j;
        sortandrank(a1,a2,n,m,j);
        tmp=a1,a1=a2,a2=tmp;
    }
    rank=a1,height=a2;
}
void calheight(char*str,int n)
{
    int i,j,k;
    sa[-1]=n;
    for(height[0]=k=i=0;i<n;i++)
    {
        for(k?k--:0,j=sa[rank[i]-1];str[i+k]==str[j+k];k++);
        height[rank[i]]=k;
    }
}
char str[maxn];
int main()
{
    int n;cin.sync_with_stdio(false);
    cin>>n;
    while(n--)
    {
        cin>>str;
        int len=strlen(str);str[len]=0;
        da(str,len+1,128);calheight(str,len);
        int ans=0;
        for(int i=1;i<=len;i++)
        {
            ans+=len-sa[i]-height[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}


0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:119161次
    • 积分:2110
    • 等级:
    • 排名:第17954名
    • 原创:84篇
    • 转载:2篇
    • 译文:0篇
    • 评论:22条
    文章分类
    最新评论