【SPOJ705】New Distinct Substrings 后缀数组

题目描述

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

题目大意

求一个字符串有多少个不同的子串。

数据范围

长度小于200000，中间可能有空格。

样例输入

ABABA

样例输出

9

解题思路

后缀数组，求出sa,rank,height。对于每一个新的后缀，则会产生它的长度个前缀，但其中是有重复的，需要减掉。height[i]=LCP(rank[i],rank[i-1])，所以答案就是

∑i=1n(n−Sai−Heighti+1)

还有就是答案会爆int 虽然这很显然

代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
inline int Getint(){int x=0,f=1;char ch=getchar();while('0'>ch||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}
namespace Suffix_Array{
    #define Suffix_Maxn 200005
    char s[Suffix_Maxn];
    int a[Suffix_Maxn],sa[2][Suffix_Maxn],rk[2][Suffix_Maxn],height[Suffix_Maxn],v[Suffix_Maxn];
    int Sa[Suffix_Maxn],Rank[Suffix_Maxn];
    int k,n,p=0,q=1;
    void Mul(int *rk,int *sa,int *RK,int *SA){
        for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;
        for(int i=n;i>=1;i--)if(sa[i]>k)SA[v[rk[sa[i]-k]]--]=sa[i]-k;
        for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
        for(int i=1;i<=n;i++)RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||(rk[SA[i]+k]!=rk[SA[i-1]+k]));
    }
    void Suffix_Init(int m){
        for(int i=1;i<=n;i++)v[a[i]]++;
        for(int i=1;i<=m;i++)v[i]+=v[i-1];
        for(int i=n;i>=1;i--)sa[p][v[a[i]]--]=i;
        for(int i=1;i<=n;i++)rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
        for(k=1;k<n;k<<=1,swap(p,q))Mul(rk[p],sa[p],rk[q],sa[q]);
        memcpy(Sa,sa[p],sizeof(sa[p]));
        memcpy(Rank,rk[p],sizeof(rk[p]));
        for(int i=1,k=0;i<=n;i++){
            int j=Sa[Rank[i]-1];
            while(a[i+k]==a[j+k])k++;
            height[Rank[i]]=k;
            if(k>0)k--;
        }
    }
}
using namespace Suffix_Array;
int main(){
    gets(s);
    n=strlen(s);
    for(int i=1;i<=n;i++)a[i]=s[i-1];
    Suffix_Init(666);
    unsigned long long Ans=0;
    for(int i=1;i<=n;i++)
        Ans+=n-Sa[i]-height[i]+1;
    cout<<Ans<<"\n";
    return 0;
}

转载于:https://www.cnblogs.com/Cedric341561/p/6811030.html