后缀数组基数排序——bzoj3172: [Tjoi2013]单词

http://www.lydsy.com/JudgeOnline/problem.php?id=3172
我靠，我那个模版没有奇数排序的；
如果直接套基数排序的话，只能模拟基数排序；
我靠，我只好用一个链表代替桶来进行基数排序；
常数是别人的4倍多

这道题直接把字符串接起来；
然后对每一个串从rank[开头]前后枚举，遇到height[i]<串长就停下；
输出答案；

#include<bits/stdc++.h>
#define rank fuck
#define Ll long long
using namespace std;
const int N=1e6+205;
struct zz{int id,x,y;}a[N];
struct cs{int nxt;zz k;}q[N];
int head[N],ll;
int x[205],y[205];
int rank[N],sa[N],h[N];
int n,m,z,ans;
char c[N],s[N];
void init(int x,zz k){q[++ll].k=k;q[ll].nxt=head[x];head[x]=ll;}
void SORT(int c){
    ll=0;for(int i=1;i<=c;i++)head[i]=0;
    for(int i=n;i;i--)init(a[i].y,a[i]);//这里要倒着放，因为倒挂
    for(int m=0,i=1;i<=c;i++)for(int k=head[i];k;k=q[k].nxt)a[++m]=q[k].k;
    ll=0;for(int i=1;i<=c;i++)head[i]=0;
    for(int i=n;i;i--)init(a[i].x,a[i]);
    for(int m=0,i=1;i<=c;i++)for(int k=head[i];k;k=q[k].nxt)a[++m]=q[k].k;
}
void make(){
    for(int i=1;i<=n;i++)a[i].x=c[i],a[i].y=0,a[i].id=i;
    SORT(127);
    for(int num=0,j=1;j<=n;j++){
        num+=(a[j].x!=a[j-1].x||a[j].y!=a[j-1].y);
        rank[a[j].id]=num;
    }
    for(int i=0;(1<<i)<=n;i++){
        for(int j=1;j<=n;j++){
            a[j].id=j;
            a[j].x=rank[j];
            a[j].y=(j+(1<<i))<=n?rank[j+(1<<i)]:0;
        }
        SORT(n);
        for(int num=0,j=1;j<=n;j++){
            num+=(a[j].x!=a[j-1].x||a[j].y!=a[j-1].y);
            rank[a[j].id]=num;
        }
    }
    for(int i=1;i<=n;i++)sa[rank[i]]=i;
    for(int i=1,k=0;i<=n;h[rank[i]]=k,i++)
        for(k?k--:k;c[i+k]==c[sa[rank[i]-1]+k];k++);
}
void init(){
    scanf("%d",&z);
    for(int i=1;i<=z;i++){
        scanf("%s",s+1);
        m=strlen(s+1);
        x[i]=n+1;
        for(int j=1;j<=m;j++)c[n+j]=s[j];
        y[i]=n=n+m;
        c[++n]=' ';
    }
}
int main()
{
    init();
    make();
    for(int i=1;i<=z;i++){
        m=y[i]-x[i]+1;ans=0;
        for(int l=rank[x[i]];h[l]>=m;l--,ans++);
        for(int r=rank[x[i]];h[r+1]>=m;r++,ans++);      
        printf("%d\n",ans+1);
    }
}