BZOJ 1692: [Usaco2007 Dec]队列变换 后缀数组 + 贪心

Description

FJ打算带他的N(1 <= N <= 30,000)头奶牛去参加一年一度的“全美农场主大奖赛”。在这场比赛中，每个参赛者都必须让他的奶牛排成一列，然后领她们从裁判席前依次走过。 今年，竞赛委员会在接受队伍报名时，采用了一种新的登记规则：他们把所有队伍中奶牛名字的首字母取出，按它们对应奶牛在队伍中的次序排成一列（比如说，如果FJ带去的奶牛依次为Bessie、Sylvia、Dora，登记人员就把这支队伍登记为BSD）。登记结束后，组委会将所有队伍的登记名称按字典序升序排列，就得到了他们的出场顺序。 FJ最近有一大堆事情，因此他不打算在这个比赛上浪费过多的时间，也就是说，他想尽可能早地出场。于是，他打算把奶牛们预先设计好的队型重新调整一下。 FJ的调整方法是这样的：每次，他在原来队列的首端或是尾端牵出一头奶牛，把她安排到新队列的尾部，然后对剩余的奶牛队列重复以上的操作，直到所有奶牛都被插到了新的队列里。这样得到的队列，就是FJ拉去登记的最终的奶牛队列。 接下来的事情就交给你了：对于给定的奶牛们的初始位置，计算出按照FJ的调整规则所可能得到的字典序最小的队列。

Input

* 第1行: 一个整数：N

* 第2..N+1行: 第i+1行仅有1个'A'..'Z'中的字母，表示队列中从前往后数第i 头奶牛名字的首字母

Output

* 第1..??行: 输出FJ所能得到的字典序最小的队列。每行（除了最后一行）输 出恰好80个'A'..'Z'中的字母，表示新队列中每头奶牛姓名的首 字母

题解： 

用后缀数组直接比较出前缀和后缀的字典序哪个更小. 

#include <bits/stdc++.h>
#define setIO(s) freopen(s".in", "r", stdin) 
#define maxn 100000
using namespace std;
int n, m, tot; 
int arr[maxn], height[maxn], A[maxn]; 
namespace SA
{
    int rk[maxn], tp[maxn], sa[maxn], tax[maxn]; 
    void qsort()
    {
        for(int i = 0; i <= m ; ++i) tax[i] = 0;
        for(int i = 1; i <= n ; ++i) ++tax[rk[i]]; 
        for(int i = 1; i <= m ; ++i) tax[i] += tax[i - 1];
        for(int i = n; i >= 1 ; --i) sa[tax[rk[tp[i]]]--] = tp[i]; 
    }
    void build()
    {
        for(int i = 1; i <= n ; ++i) rk[i] = arr[i], tp[i] = i;
        qsort(); 
        for(int k = 1; k <= n ; k <<= 1)
        {
            int p = 0;
            for(int i = n - k + 1; i <= n ; ++i) tp[++p] = i; 
            for(int i = 1; i <= n ; ++i) if(sa[i] > k) tp[++p] = sa[i] - k;
            qsort(), swap(rk, tp), rk[sa[1]] = p = 1;
            for(int i = 2; i <= n ; ++i)
            {
                rk[sa[i]] = (tp[sa[i - 1]] == tp[sa[i]] && tp[sa[i - 1] + k] == tp[sa[i] + k]) ? p : ++p; 
            }
            if(n == p) break; 
            m = p; 
        }
        int k = 0;
        for(int i = 1; i <= n ; ++i) rk[sa[i]] = i;
        for(int i = 1; i <= n ; ++i)
        {
            if(k) --k;
            int j = sa[rk[i] - 1]; 
            while(arr[i + k] == arr[j + k]) ++k;
            height[rk[i]] = k; 
        }  
    }
};
int main()
{
    // setIO("input"); 
    int len; 
    char str[2]; 
    scanf("%d",&len);
    for(int i = 1; i <= len; ++i)  scanf("%s", str), arr[++n] = A[i] = (int)(str[0] - 'A'); 
    arr[++n] = 0;        
    for(int i = len ; i >= 1 ; --i) arr[++n] = A[i];
    m = 130;  
    SA::build();
    int l = 1, r = len + 2;  
    for(int i = 1; i <= len ; ++i)
    {   
        if(SA::rk[l] < SA::rk[r])
        {
            printf("%c",(char)arr[l] + 'A'), ++l; 
        }
        else 
        {
            printf("%c",(char)arr[r] + 'A'), ++r;  
        }
        if(i % 80 == 0) printf("\n"); 
    }
    return 0; 
}