后缀数组模板(详细理解)

原创 2016年08月29日 20:14:12

后缀数组

后缀数组看了很久了,从基数排序开始看一只看到height数组的实现也理解了很多东西,不过别人的东西再多也不是自己的,因此自己重新总结了一下,顺便学一个后缀数组的模板备用,这里以SPOJ 694为例。

#include <ctime>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+5;

char s[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int rank[maxn],height[maxn];

void build_sa(int n,int m){
    int i,*x = t, *y = t2;  //引用指针只是为了后面好交换
    for(i = 0; i < m; i++) c[i] = 0;
    for(i = 0; i < n; i++) c[x[i] = s[i]]++;
    for(i = 1; i < m; i++) c[i] += c[i-1];
    for(i = n-1; i >= 0; i--) sa[--c[x[i]]] = i;  //sa[i]中表示排名第i的位置是多少
    for(int k = 1; k <= n; k <<= 1){ //k表示每次基数排序需要比较的长度,因为是按照倍增算法所以每次比较2个关键字
        int p = 0;
        //直接利用sa数组排序第二关键字
        for(i = n-k; i < n; i++) y[p++] = i; //y中存放按第二关键字从小到大排序的位置
        for(i = 0; i < n; i++) if(sa[i] >= k) y[p++] = sa[i]-k;
        //基数排序第一关键字
        for(i = 0; i < m; i++) c[i] = 0;
        for(i = 0; i < n; i++) c[x[y[i]]]++;
        for(i = 0; i < m; i++) c[i] += c[i-1];
        for(i = n-1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i];   //i从大到小是为了保证相同字符的情况下默认靠前的更小一些
        swap(x, y);  //这里只用交换指针即可
        p = 1; x[sa[0]] = 0; //p表示rank值不同的字符串的数量,如果达到n表示字符串的所有关系都找出来了
        for(i = 1; i < n; i++)  //重新计算x的值
            x[sa[i]] = y[sa[i-1]] == y[sa[i]] && y[sa[i-1]+k] == y[sa[i]+k]?p-1:p++;
        if(p >= n) break;
        m = p;
    }
}

void getHeight(int n){
    int i,j,k = 0;
    for(i = 0; i < n; i++) rank[sa[i]] = i; //求出rank值,利用rank和sa是相反的
    for(i = 0; i < n; i++){
        if(k) k--; //利用h[i] >= h[i-1]+1这个性质,先求出前面的后面的就可以由前面推出
        j = sa[rank[i]-1];
        while(s[i+k] == s[j+k]) k++;
        height[rank[i]] = k;
    }
}

int main(){
    #ifdef LOCAL
        freopen("input.txt", "r", stdin);
        freopen("output.txt", "w", stdout);
        int _time_jc = clock();
    #endif // LOCAL
    int T;
    scanf("%d", &T);
    while(T--){
        scanf("%s", s);
        int len = strlen(s);
        build_sa(len+1, 130);  //为了添加一个由字符串结束符为后缀的字符串
        getHeight(len+1);
//      printf("%d\n", sa[0]);
        int ans = 0;
        for(int i = 1; i <= len; i++)
            ans += (len-sa[i]-height[i]);
        printf("%d\n", ans);
    }
    #ifdef LOCAL
        printf("time: %d\n", int(clock() - _time_jc));
    #endif // LOCAL
    return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

后缀数组模板(理解)

字符串的处理真可谓是博大精深,后缀数组这种数据结构我花了两天时间才明白了其构造的过程。主要是代码不好理解。 数据结构: 1.sa数组,就是后缀数组,按照字典序排列,其意义为:sa[i]=k,排第i...

后缀数组PPT详细解答

  • 2009-07-25 00:18
  • 419KB
  • 下载

[Tyvj 1860] 后缀数组 倍增O(nlogn) 求sa[],height[]模板,单调栈维护,求特征值

#include #include #include using namespace std; const int maxn=200005; char text[maxn]; int SA[maxn]...

后缀数组(dc3 +RMQ线段树维护模板)

#include #include #include #include #define maxn 200100 using namespace std; int r[maxn]; int Rank[m...

后缀数组模板

刘汝佳的代码会RE!!!

SA(后缀数组)——倍增法模板

倍增法原理:一步一步把n个后缀按照它们的前 2 ^ k 的字符排序。

后缀数组 模板

搞了一下午才把后缀数组稍微明白了点  真是弱呀~~ 根据网上各种大神的模板  找了一份自己好看的 留着以后用.... #include #include const int maxn =2000...
  • Taceo
  • Taceo
  • 2013-08-21 17:17
  • 555

后缀数组 相关问题模板

后缀数组函数模板: const int MAX=200020; int wa[MAX],wb[MAX],wsf[MAX],wv[MAX],sa[MAX]; int rank[MAX],height[M...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)