码迷,mamicode.com
首页 > 编程语言 > 详细

后缀数组 POJ 3581 Sequence

时间:2016-04-29 23:29:18      阅读:247      评论:0      收藏:0      [点我收藏+]

标签:

 

题目链接

题意:把n个数字(A1比其他数字都大)的序列分成三段,每段分别反转,问字典序最小的序列。

分析:因为A1比其他数字都大,所以反转后第一段结尾是很大的数,相当是天然的分割线,第一段可以单独考虑,即求整段的字典序最小的后缀。后面两段不能分开考虑,

例子:

9
8 4 -1 5 0 5 0 2 3
第一步:
3 2 0 5 0 5 -1 4 8 对应输出 -1 4 8
第二步
3 2 0 5 0 5(开始的时候我并没有复制一遍) 对应输出:0 5
第三步
3 2 0 5    对应输出: 3 2 0 5
可以看见这样做是不对的。。
必须要将剩下的字符串复制一遍贴在后面,然后再来求后缀数组。。。
正解:
第一步:
3 2 0 5 0 5 -1 4 8 对应输出 -1 4 8
第二步
3 2 0 5 0 5 3 2 0 5 0 5 对应输出: 0 5 0 5;
第三步
3 2 对应输出:3 2;

所以方法是剩下的反转后+剩下的反转后组成新的串,求sa(这里不用求height,只要扫一遍即可),找到符合条件的字典序最小的后缀(应该是长度为剩下的长度的前缀)。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>

typedef long long ll;
const int N = 2e5 + 5;
int a[N];
int rev[N<<1];
int sa[N<<1], rank[N<<1];
int tmp[N<<1];
int n, k;

bool cmp_sa(int i, int j) {
    if (rank[i] != rank[j]) {
        return rank[i] < rank[j];
    } else {
        int ri = i + k <= n ? rank[i+k] : -1;
        int rj = j + k <= n ? rank[j+k] : -1;
        return ri < rj;
    }
}

void get_sa(int *a, int n, int *sa) {
    for (int i=0; i<=n; ++i) {
        sa[i] = i;
        rank[i] = i < n ? a[i] : -1;
    }
    for (k=1; k<=n; k<<=1) {
        std::sort (sa, sa+n+1, cmp_sa);
        tmp[sa[0]] = 0;
        for (int i=1; i<=n; ++i) {
            tmp[sa[i]] = tmp[sa[i-1]] + (cmp_sa (sa[i-1], sa[i]) ? 1 : 0);
        }
        for (int i=0; i<=n; ++i) {
            rank[i] = tmp[i];
        }
    }
}

void run() {
    std::reverse_copy (a, a+n, rev);
    get_sa (rev, n, sa);
    int p1;
    for (int i=0; i<n; ++i) {
        p1 = n - sa[i];
        if (p1 >= 1 && sa[i] >= 2) {
            break;
        }
    }
    int m = n - p1;
    std::reverse_copy (a+p1, a+n, rev);
    std::reverse_copy (a+p1, a+n, rev+m);
    get_sa (rev, m*2, sa);
    int p2;
    for (int i=0; i<=m*2; ++i) {
        p2 = p1 + m - sa[i];
        if (p2 - p1 >= 1 && p2 < n) {
            break;
        }
    }
    std::reverse (a, a+p1);
    std::reverse (a+p1, a+p2);
    std::reverse (a+p2, a+n);
    for (int i=0; i<n; ++i) {
        printf ("%d\n", a[i]);
    }
}

int main() {
    scanf ("%d", &n);
    for (int i=0; i<n; ++i) {
        scanf ("%d", a+i);
    }
    run ();
    return 0;
}

  

 

后缀数组 POJ 3581 Sequence

标签:

原文地址:http://www.cnblogs.com/Running-Time/p/5447734.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!