SPOJ705 Distinct Substrings （后缀自动机&后缀数组）

Given a string, we need to find the total number of its distinct substrings.

Input

T- number of test cases. T<=20;
Each test case consists of one string, whose length is <= 1000

Output

For each test case output one number saying the number of distinct substrings.

Example

Sample Input:
2
CCCCC
ABABA

Sample Output:
5
9

Explanation for the testcase with string ABABA: 
len=1 : A,B
len=2 : AB,BA
len=3 : ABA,BAB
len=4 : ABAB,BABA
len=5 : ABABA
Thus, total number of distinct substrings is 9.

题意：

求出大写的字符串里不同的子串。默写了一遍后缀自动机。今天主要是练习后缀数组。

注意：

• 注意是大写还是小写；
• 注意init初始化的时候没有一次性memset，所以下面要把每个新出现的点memset。不要搞忘。

后缀自动机：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=;
struct SAM
{
    int ch[maxn][],fa[maxn],maxlen[maxn],Last,sz;
    void init()
    {
        sz=Last=;    fa[]=maxlen[]=;
        memset(ch[],,sizeof(ch[]));
    }
    void add(int x)
    {
        int np=++sz,p=Last;Last=np;
        memset(ch[np],,sizeof(ch[np]));
        maxlen[np]=maxlen[p]+;
        while(p&&!ch[p][x]) ch[p][x]=np,p=fa[p];
        if(!p) fa[np]=;
        else {
            int q=ch[p][x];
            if(maxlen[p]+==maxlen[q]) fa[np]=q;
            else {
                int nq=++sz;
                memcpy(ch[nq],ch[q],sizeof(ch[q]));
                maxlen[nq]=maxlen[p]+;
                fa[nq]=fa[q];
                fa[q]=fa[np]=nq;
                while(p&&ch[p][x]==q) ch[p][x]=nq,p=fa[p];
            }
        }
    }
};
SAM Sam;
int main()
{
    char chr[maxn];
    int T,ans,i,L;
    scanf("%d",&T);
    while(T--){
        Sam.init();ans=;
        scanf("%s",chr);
        L=strlen(chr);
        for(i=;i<L;i++) Sam.add(chr[i]-'A');
        for(i=;i<=Sam.sz;i++) ans+=Sam.maxlen[i]-Sam.maxlen[Sam.fa[i]];
        printf("%d\n",ans);
    }
    return ;
} 

后缀数组：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=;
char ch[maxn];
int L;
struct SA
{
    int cntA[maxn],cntB[maxn],A[maxn],B[maxn];
    int rank[maxn],sa[maxn],tsa[maxn],ht[maxn];void sort()
    {
         for (int i = ; i < ; i ++) cntA[i] = ;
         for (int i = ; i <= L; i ++) cntA[ch[i]] ++;
         for (int i = ; i < ; i ++) cntA[i] += cntA[i - ];
         for (int i = L; i; i --) sa[cntA[ch[i]] --] = i;
         rank[sa[]] = ;
         for (int i = ; i <= L; i ++){
              rank[sa[i]] = rank[sa[i - ]];
              if (ch[sa[i]] != ch[sa[i - ]]) rank[sa[i]] ++;
         }
         for (int l = ; rank[sa[L]] < L; l <<= ){
              for (int i = ; i <= L; i ++) cntA[i] = ;
              for (int i = ; i <= L; i ++) cntB[i] = ;
              for ( int i = ; i <= L; i ++){
                  cntA[A[i] = rank[i]] ++;
                  cntB[B[i] = (i + l <= L) ? rank[i + l] : ] ++;
              }
              for (int i = ; i <= L; i ++) cntB[i] += cntB[i - ];
              for (int i = L; i; i --) tsa[cntB[B[i]] --] = i;
              for (int i = ; i <= L; i ++) cntA[i] += cntA[i - ];
              for (int i = L; i; i --) sa[cntA[A[tsa[i]]] --] = tsa[i];
              rank[sa[]] = ;
              for (int i = ; i <= L; i ++){
                   rank[sa[i]] = rank[sa[i - ]];
                   if (A[sa[i]] != A[sa[i - ]] || B[sa[i]] != B[sa[i - ]]) rank[sa[i]] ++;
              }
         }
    }
    void getheight()
    {
         for (int i = , j = ; i <= L; i ++){
              if (j) j --;
              while (ch[i + j] == ch[sa[rank[i] - ] + j]) j ++;
              ht[rank[i]] = j;
        }
    }
};
SA Sa;
int main()
{
    int T,ans,i;
    scanf("%d",&T);
    while(T--){
        ans=;
        scanf("%s",ch+);
        L=strlen(ch+);
        Sa.sort();
        Sa.getheight();
        for(i=;i<=L;i++) ans+=L-Sa.sa[i]+-Sa.ht[i];
        printf("%d\n",ans);
    }
    return ;
}