spoj 后缀数组

来源:互联网 发布:淘宝什么最好卖 编辑:程序博客网 时间:2024/06/14 04:05

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;//就这里
0 0