poj 3581 Sequence（后缀数组，离散化）详解

题目链接：http://poj.org/problem?id=3581

题目大意：给一个数列，要求将其分成三段，每段进行翻转后形成后合并成新数列，求按字典顺序最小的新数列。

思路：

    注意到题目中数列a0,a2,a3...an-1, a0是最大的，因此将原数列翻转后an-1,an-2,...,a1,a0,求后缀数组，

    sa[0]所代表的后缀即为所求第一段翻转后的数列，注意到要分成三份，因此sa[0]<2时不可取，此时找sa[1],

    sa[2]看是否可取。找第一个位置后，设剩下 数列是an-1,an-2,...,ak, 问题转化为找一个分割位置m，使得

    am,..ak, an-1,an-2,...,am+1,字典顺序最小。因此将原数列扩展成an-1,an-2,...,ak,an-1,an-2,...,ak,求后缀数组

    这样，找到一个最小的i, 使得sa[i]>0, sa[i]<n-k(即要在前一部分），则m=sa[i]. 此时后缀sa[i]的前n-k个前缀刚好是

    要求的翻转后的第二三部分。

    另外就是要进行离散化，但要保证原数列之间的相对大小关系。

详细代码：

#include <map>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define rank rk
using namespace std;

const int maxn=200010;
int s[maxn];
int n;
int sa[maxn],t[maxn],t2[maxn],c[maxn];
// 后缀数组模板
void build_sa(int m){
    int *x=t, *y=t2;
    memset(c, 0, sizeof(int)*m);
    for(int i=0; i<n; ++i) c[x[i]=s[i]]++;
    for(int i=1; i<m; ++i) c[i]+=c[i-1];
    for(int i=n-1; i>=0; --i) sa[--c[x[i]]]=i;
    for(int k=1; k<=n; k<<=1){
        int p=0;
        for(int i=n-k; i<n; ++i) y[p++]=i;
        for(int i=0; i<n; ++i) 
            if(sa[i]>=k) y[p++]=sa[i]-k;
        memset(c, 0, sizeof(int)*m);
        for(int i=0; i<n; ++i) c[x[y[i]]]++;
        for(int i=1; i<m; ++i) c[i] += c[i-1];
        for(int i=n-1; i>=0; --i) sa[--c[x[y[i]]]]=y[i];
        std::swap(x,y);
        y[n]=-1;
        p=1; x[sa[0]]=0;
        for(int i=1; i<n; ++i)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        m=p;
        if(p>=n) break;
    }
}
// 用于离散化
map<int,int> src2i, i2src;
map<int,int>::iterator it;

void print(int i, int t){
    while(i<t){
        printf("%d
", i2src[s[i++]]);
    }
}
int main(){
    scanf("%d", &n);
    for(int i=0; i<n; ++i){
        scanf("%d", &t[i]);
        src2i[t[i]]=1;
    }
    int cnt=1;
    for(it=src2i.begin(); it!=src2i.end(); ++it){
        i2src[cnt]=it->first;
        it->second=cnt++;
    }
    for(int i=0; i<n; ++i)
        s[n-1-i]=src2i[t[i]];
    build_sa(maxn);
    int idx1=sa[0], i=1;
    while(idx1<2) idx1=sa[i++];
    print(idx1, n);
    copy(s, s+idx1, s+idx1);// 扩展数组
    n=2*idx1;    build_sa(maxn);
    int idx2=sa[0]; i=1;
    while(idx2>=idx1 || idx2<1) idx2=sa[i++];
    print(idx2, idx2+idx1);
    return 0;
}