G - Distinct Substrings SPOJ - DISUBSTR

 https://cn.vjudge.net/contest/311974#problem/G

题意：求不同子串数量；

思路：减掉height[i]是减掉以sa[i]开头的串的数量；无需减掉它的子串；因为之后会处理掉（它的子串sa[i]不同） 

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;

#define sfi(x) scanf("%d",&x)
#define sfc(x) scanf("%c",x)
#define sfl(x) scanf("%lld",&x)
#define sfs(x) scanf("%s",x)

#define rint register int
#define pb push_back
#define fl() printf("flag!\n")
#define INF 0x3f3f3f3f
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

const int maxn=1e6+9;
const int mod=1e9+7;

char s[maxn];
int b[maxn];
int t1[maxn],t2[maxn],rk[maxn],sa[maxn],c[maxn],height[maxn];
int n,m;

void get_sa(char *s,int n,int m)
{
    int *x=t1,*y=t2;
    int i,k;
    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)
    {
        int 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);
        x[sa[0]]=0;
        p=1;
        for(i=1;i<n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) ? p-1:p++;
        if(p>=n) break;
        m=p;
    }
}


void get_height(char *s,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rk[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k) --k;
        else k=0;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}

void debug(int n)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=sa[i];j<n;j++) cout<<s[j];
        cout<<endl;
    }
}


int main()
{
    //FAST_IO;
    //freopen("input.txt","r",stdin);

    int T;
    sfi(T);
    while(T--)
    {
        sfs(s);
        n=strlen(s);
        s[n]=0;
        m=300;

        //cout<<s<<endl;

        get_sa(s,n+1,m);
        get_height(s,n);
        //debug(n);


        ll ans=(n+1)*(n)/2;
        for(int i=1;i<=n;i++)
        {
            if(height[i]>0)
            {
                ans-=height[i];
            }
        }

        printf("%lld\n",ans);


    }


    return 0;

}