POJ 3581 Sequence

首先将字符串翻转，求后缀数组，得到一个最小后缀，当然这个最小后缀的起始点应该在原串的倒数第二个位置之前，不然无法分成三段 
然后对于剩下的字符串，相当于是要分成2份然后字典序最小，对于这个，我一开始想用最小表示法的那个线性方法来求，后来发现这样可能会出现不切分的情况，也就是这个字符串的字典序已经是最小的了 
于是我把剩余字符串重复一次接到一起，然后再求一次后缀数组，这样就求出了每一个字符串表示法的字典序排名，当然我们要的是字典序最小的而且位置在字符串倒数第一个位置之前的

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#include <string>
#define max_n 200000+100
using namespace std;
int n,k;
int rank1[max_n+1];
int temp[max_n+1];

bool cmp(int i,int j)//k是全局变量，我在下面重新定义了一次，找了两个小时的bug
{
if(rank1[i]!=rank1[j])
return rank1[i]<rank1[j];
else
{
int ri=i+k<=n?rank1[i+k]:-1;
int rj=j+k<=n?rank1[j+k]:-1;
return ri<rj;
}
}
void construct_sa(int *s,int N,int *sa)
{
n=N;
for(int i=0; i<=n; i++) {
sa[i] = i;
rank1[i] = i<n?s[i]:-1;
}
for(k=1; k<=n; k*=2)
{
sort(sa, sa+n+1, cmp);
temp[sa[0]] = 0;
for(int i=1; i<=n; i++)
temp[sa[i]] = temp[sa[i-1]] + (cmp(sa[i-1], sa[i])?1:0);
for(int i=0; i<=n; i++) rank1[i] = temp[i];
}
}
int N,A[max_n];
int rev[max_n],sa[max_n];
void solve()
{
reverse_copy(A,A+N,rev);
construct_sa(rev,N,sa);
int p1;
for(int i=0;i<N;i++)
{
p1=N-sa[i];
if(p1>=1&&N-p1>=2)break;
}
int m=N-p1;
reverse_copy(A+p1,A+N,rev);
reverse_copy(A+p1,A+N,rev+m);
construct_sa(rev,2*m,sa);
int p2;
for(int i=0;i<=2*m;i++)
{
p2=p1+m-sa[i];
if(p2-p1>=1&&N-p2>=1)break;
}
reverse(A,A+p1);
reverse(A+p1,A+p2);
reverse(A+p2,A+N);
for(int i=0;i<N;i++)
printf("%d\n",A[i]);
}
int main(){
cin>>N;
for(int i=0;i<N;i++)
scanf("%d",&A[i]);
solve();
return 0;
}