后缀数组

  没错 抄的板子 理不壮,气也直! 

所以.....我把这个优秀的博主链接贴上 hhhhhhhhhhhh  (博客也很漂亮)

https://www.cnblogs.com/victorique/p/8480093.html

 

class SuffixArray {
    
    char s[maxn];  // 注意 s是[1,n]的 所以用字符长度的时候也要注意+1等问题. 
    int t1[maxn],t2[maxn],c[maxn],sa[maxn],rk[maxn],height[maxn];
    //  t1 t2 c 是临时数组
    //  rk 是 名次数组  rank[i] 第i位排名第几     ---> 你排老几
    //  sa 是 后缀数组  sa[i]   排第i名的起始位置 ---> 老几你在哪儿 
    int cn,cm; // n是字符串长度  m 是最大的字符值 
    
    void init()
    {
        memset(c, 0, sizeof(c));    // 经过大概的测试..桶需要初始化. 
    }
    
    void get_SA() {  // [1...n] 
        init();
        int i, k;
        int *x = t1, *y = t2;
        for (i=1; i<=cn; ++i) ++c[x[i]=s[i]];
            //c数组是桶
            //x[i]是第i个元素的第一关键字
        for (i=2; i<=cm; ++i) c[i]+=c[i-1];
            //做c的前缀和,我们就可以得出每个关键字最多是在第几名
        for (i=cn; i>=1; --i) sa[c[x[i]]--]=i;
        for (k=1; k<=cn; k<<=1) {
            int num=0;
            for (i=cn-k+1; i<=cn; ++i) y[++num]=i;
            //y[i]表示第二关键字排名为i的数,第一关键字的位置
            //第n-k+1到第n位是没有第二关键字的 所以排名在最前面
            for (i=1; i<=cn; ++i) if (sa[i]>k) y[++num]=sa[i]-k;
            //排名为i的数 在数组中是否在第k位以后
            //如果满足(sa[i]>k) 那么它可以作为别人的第二关键字,就把它的第一关键字的位置添加进y就行了
            //所以i枚举的是第二关键字的排名,第二关键字靠前的先入队
            for (i=1; i<=cm; ++i) c[i]=0;
            //初始化c桶
            for (i=1; i<=cn; ++i) ++c[x[i]];
            //因为上一次循环已经算出了这次的第一关键字 所以直接加就行了
            for (i=2; i<=cm; ++i) c[i]+=c[i-1]; //第一关键字排名为1~i的数有多少个
            for (i=cn; i>=1; --i) sa[c[x[y[i]]]--]=y[i],y[i]=0;
            //因为y的顺序是按照第二关键字的顺序来排的
            //第二关键字靠后的,在同一个第一关键字桶中排名越靠后
            //基数排序
            swap(x,y);
            //这里不用想太多,因为要生成新的x时要用到旧的,就把旧的复制下来,没别的意思 这儿用指针优化了下. 
            x[sa[1]]=1;
            num=1;
            for (i=2; i<=cn; ++i)
                x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num;
            //因为sa[i]已经排好序了,所以可以按排名枚举,生成下一次的第一关键字
            if (num==cn) break;
            cm=num;
            //这里就不用那个122了,因为都有新的编号了
        }
    }
    
    void get_height() {  // [1...n] 
        int i, j, k=0;
        for (i=1; i<=cn; ++i) rk[sa[i]]=i;
        for (i=1; i<=cn; ++i) {
            if (rk[i]==1) continue;//第一名height为0
            if (k) --k;//h[i]>=h[i-1]+1;
            j=sa[rk[i]-1];
            while (j+k<=cn && i+k<=cn && s[i+k]==s[j+k]) ++k;
            height[rk[i]]=k;//h[i]=height[rk[i]];
        }
        height[0] = 0;
    }
}; 

 

posted @ 2018-07-30 10:37  过路人1998  阅读(87)  评论(0编辑  收藏  举报