spoj 后缀数组

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=115232#problem/B

求给出字符串所有的不重复子串，可以重叠，

所有后缀的所有前缀就包括了所有字符串 用总字符串的数去减重复字符串的个数 即 所有height【】，

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn=100010;

int wa[maxn], wb[maxn], wv[maxn], ws1[maxn],sa[maxn];
int cmp(int *r, int a, int b, int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m)
{
    int i, j, p, *x = wa,*y = wb;
    // 下面四行是对第一个字母的一个基数排序：基数排序其实就是记录前面有多少个位置被占据了
    for(i = 0; i<m; i++) ws1[i]=0; // 将统计字符数量的数组清空
    for(i = 0; i<n; i++) ws1[x[i]=r[i]]++; // 统计各种字符的个数
    for(i = 1; i<m; i++) ws1[i]+=ws1[i-1]; // 进行一个累加，因为前面的小字符集对后面字符的排位有位置贡献
    for(i = n-1; i>=0; i--) sa[--ws1[x[i]]]=i; // 根据位置来排序，sa[x] = i，表示i位置排在第x位
    // wa[x[i]]就是字符集0-x[i]共有多少字符占据了位置，减去自己的一个位置剩下的就是自己的排名了，排名从0开始
    // 排名过程中主要的过程是对于处于相同字符的字符的排序，因为改变wa[x[i]]值得只会是本身，小于该字符的贡献值
    // 是不变的，对于第一个字符相同的依据是位置关系，在后面将看到通过第二个关键字来确定相同字符的先后关系


    // 这以后的排序都是通过两个关键字来确定一个串的位置，也即倍增思想
    // 通过将一个串分解成两部分，而这两部分的位置关系我们都已经计算出来
    for(j = 1, p = 1; p<n; j*=2, m=p)
    {
        for(p = 0, i = n-j; i<n; i++) y[p++]=i; // 枚举的串是用于与i位置的串进行合并，由于i较大，因为匹配的串为空串
        // 由于枚举的是长度为j的串，那么i位置开始的串将凑不出这个长度的串，因此第二关键字应该最小，这其中位置靠前的较小
        for(i = 0; i<n; i++)
            if(sa[i]>=j) y[p++]=sa[i]-j; // sa[i]-j开头的串作为第二关键字与编号为sa[i]的串匹配，sa[i]<j的串不用作为第二关键字来匹配
        for(i = 0; i<n; i++) wv[i]=x[y[i]]; // 取出这些位置的第一关键字
        for(i = 0; i<m; i++) ws1[i]=0;
        for(i = 0; i<n; i++) ws1[wv[i]]++;
        for(i = 1; i<m; i++) ws1[i]+=ws1[i-1];
        for(i = n-1; i>=0; i--) sa[--ws1[wv[i]]]=y[i]; // 按照第二关键字进行第一关键字的基数排序
        for(swap(x,y),p=1,x[sa[0]]=0,i=1; i<n; i++) // 对排好序的sa数组进行一次字符集缩小、常数优化
            x[sa[i]] = cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
    return;
}

int rank[maxn],height[maxn];
void calheight(int *r,int *sa,int n) // 这里的n是原串的本来长度，即不包括新增的0
{
    int i,j,k=0;
    for(i = 1; i<=n; i++) rank[sa[i]]=i; // 有后缀数组得到名次数组，排名第0的后缀一定是添加的0
    for(i = 0; i<n; height[rank[i++]]=k) // 以 i 开始的后缀总能够从以 i-1 开始的后缀中继承 k-1 匹配项出来
        for(k?k--:0, j=sa[rank[i]-1]; r[i+k] == r[j+k]; k++); // 进行一个暴力的匹配，但是整个算法的时间复杂度还是O(n)的
    return;
}


char s[maxn];
int ss[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s);
        int n=strlen(s);
        for(int i=0;i<=n;i++) ss[i]=s[i];
        da(ss,sa,n+1,128);
        calheight(ss,sa,n);
        int ans=n*(n+1)/2;
        for(int i=2;i<=n;i++)ans-=height[i];
        printf("%d\n",ans);
    }
    return 0;
}

705 基本一样就是范围大了 改成longlong

long long ans=(long long)n*(n+1)/2;//就这里