后缀数组SA

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f;
const int maxn=1000005;
char s[maxn];
int n,w,sa[maxn],rk[maxn<<1|1],oldrk[maxn<<1|1];
bool cmp(int x,int y){
    return rk[x]==rk[y]?rk[x+w]<rk[y+w]:rk[x]<rk[y];
    //以rank[i]为第一关键字，rank[i+w]为第二关键字
}
int main(){
    scanf("%s",s+1);
    n=strlen(s+1);
    for(int i=1;i<=n;++i)sa[i]=i;//随便给个1-n的排序
    for(int i=1;i<=n;++i)rk[i]=s[i];//先按照第一个字母排个rank，只需要相对大小即可
    for(w=1;w<n;w<<=1){
        sort(sa+1,sa+n+1,cmp);
        for(int i=1;i<=n;++i)oldrk[i]=rk[i];
        for(int p=0,i=1;i<=n;++i){
            if(oldrk[sa[i]]==oldrk[sa[i-1]]&&oldrk[sa[i]+w]==oldrk[sa[i-1]+w])rk[sa[i]]=p;
            else rk[sa[i]]=++p;
            //判断条件和p是为了去重
        }
    }
    for(int i=1;i<=n;++i)printf("%d ",sa[i]);
    return 0;
}