注意常数优化(优化输出等)。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=2e5+10;
int n,sa[MAXN],tmp[MAXN],rank[MAXN],sec[MAXN],cnt[MAXN],height[MAXN];
char s[MAXN];
void get_sa()
{
for(int i=1;i<=n;i++) rank[i]=s[i]-'a'+1;
for(int i=1;i<=n;i++) cnt[rank[i]]++;
for(int i=1;i<=26;i++)cnt[i]+=cnt[i-1];
for(int i=1;i<=n;i++) sa[cnt[rank[i]]--]=i;
for(int k=1;k<=n;k<<=1)
{
int top=0;
for(int i=n-k+1;i<=n;i++) sec[++top]=i;
for(int i=1;i<=n;i++) if(sa[i]>k) sec[++top]=sa[i]-k;
for(int i=0;i<=n;i++) cnt[i]=0;
for(int i=1;i<=n;i++) cnt[rank[i]]++;
for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;i--) sa[cnt[rank[sec[i]]]--]=sec[i];
top=tmp[sa[1]]=1;
for(int i=2;i<=n;i++)
{
if(rank[sa[i]]!=rank[sa[i-1]]||rank[sa[i]+k]!=rank[sa[i-1]+k]) top++;
tmp[sa[i]]=top;
}
for(int i=1;i<=n;i++) rank[i]=tmp[i];
}
int k=0;
for(int i=1;i<=n;i++)
{
if(k) k--;
while(i+k<=n&&sa[rank[i]-1]+k<=n&&s[i+k]==s[sa[rank[i]-1]+k]) k++;
height[rank[i]]=k;
}
}
void putt(int x)
{
if(!x)return;
char c=x%10+'0';
putt(x/10);
putchar(c);
}
int main()
{
scanf("%s",s+1);
n=strlen(s+1);
get_sa();
for(int i=1;i<=n;i++)
{
putt(sa[i]);
putchar(' ');
}
puts("");
for(int i=1;i<=n;i++)
{
if(!height[i]) putchar('0');
else putt(height[i]);
putchar(' ');
}
return 0;
}