티스토리 뷰

알고리즘 관련/BOJ

BOJ)1849 순열

JASON 자손9319 2017. 1. 10. 16:51

문제: icpc.me/1849


(A[i]=i보다 큰 수들의 개수)일 때 원래 수열을 구하는 프로그램을 작성하면 된다.


작은 수 부터 채워 나간다고 할 때 A[1]이 5 라면 1은 수열의 6번째 수가 될 것이다.


우리는 수열을 세그먼트 트리를 이용하여 구해보자.


먼저 리프에 해당하는 모든 노드를 1로 채워준 후 구간합을 업데이트 시켜준다.


이후 a[i]를 확인 할 때 a[i]+1번째에 해당하는 노드를 찾아준 뒤 수열의 해당하는 자리에 i를 저장해 준다.


이후 노드 i를 0으로 초기화 시켜준다. 이런 방식으로 나보다 큰 수들중에서만 내가 몇번째 위치에 있는지 확인 하여 수열을 채워 나간다.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#include <cstdio>
#include <algorithm>
#define MAX_N 100000
using namespace std;
int n, seg[* MAX_N], x, arr[MAX_N + 1];
int update(int pos, int val, int node, int x, int y) {
    if (y < pos || pos < x)
        return seg[node];
    if (x == y)
        return seg[node] = val;
    int mid = (x + y) >> 1;
    return seg[node] = update(pos, val, node * 2, x, mid) + update(pos, val, node * + 1, mid + 1, y);
}
int query(int val, int node, int x, int y) {
    int mid = (x + y) >> 1;
    if (x == y)
        return x;
    if (seg[node * 2>= val)
        return query(val, node * 2, x, mid);
    return query(val - seg[node * 2], node * + 1, mid + 1, y);
}
int main() {
    scanf("%d"&n);
    for (int i = 1; i <= n; i++)
        update(i, 111, n);
    for (int i = 1; i <= n; i++) {
        scanf("%d"&x);
        int q = query(x + 111, n);
        arr[q] = i;
        update(q, 011, n);
    }
    for (int i = 1; i <= n; i++)
        printf("%d\n", arr[i]);
    return 0;
}
cs


'알고리즘 관련 > BOJ' 카테고리의 다른 글

BOJ)6549 히스토그램에서 가장 큰 직사각형  (0) 2017.01.10
BOJ)5419 북서풍  (7) 2017.01.10
BOJ)1849 순열  (0) 2017.01.10
BOJ)4195 친구 네트워크  (0) 2017.01.10
BOJ)9345 디지털 비디오 디스크(DVDs)  (0) 2017.01.10
BOJ)1321 군인  (1) 2017.01.10
댓글
댓글쓰기 폼