SPOJ SUBST1 New Distinct Substrings（后缀数组）

最新推荐文章于 2019-08-18 11:03:58 发布

原创最新推荐文章于 2019-08-18 11:03:58 发布 · 279 阅读

0 ·

CC 4.0 BY-SA版权

SPOJ 同时被 2 个专栏收录

12 篇文章

订阅专栏

后缀数组

5 篇文章

订阅专栏

本文介绍了一种使用后缀数组解决求字符串中不同子串数量问题的方法。通过构造后缀数组并计算高度数组，可以高效地得出答案。

题意：

一个字符串有多少个不同的字串

思路：

题意与思路和 SPOJ DISUBSTR完全相同，唯一不同是数据范围，用后缀数组复杂度完全够，注意下long long 即可

错误及反思：

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=50000+10;
int sa[N],rak[N],height[N];
char s[N];
void construct(const char *s,int n,int m = 256)
{
    static int t1[N],t2[N],c[N];
    int *x = t1,*y = t2;
    int i,j,k,p,l;
    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;
    for (k = 1; k <= n; k <<= 1) {
        p = 0;
        for (i = n - k; i < n; ++ i) y[p++] = i;
        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 = 1; i < m; ++ i) c[i] += c[i - 1];
        for (i = n - 1; i >= 0; -- i) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        p = 1; x[sa[0]] = 0;
        for (i = 1; i < n; ++ i)
            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;
    }
    for (i = 0; i < n; ++ i) rak[sa[i]] = i;
    for (i = 0,l = 0; i < n; ++ i) {
        if (rak[i]) {
            j = sa[rak[i] - 1];
            while (s[i + l] == s[j + l]) l++;
            height[rak[i]] = l;
            if (l) l--;
        }
    }
}

int main(){
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s);
        int len=strlen(s);
        construct(s,len+1);
        long long sum=0;
        for(int i=2;i<=len;i++){
            sum+=1ll*height[i];
        }

        printf("%I64d\n",1ll*len*(len+1)/2-sum);

    }
}