[C++] 백준 1015번:수열 정렬

문제링크

1015번: 수열 정렬

문제

P[0], P[1], ...., P[N-1]은 0부터 N-1까지(포함)의 수를 한 번씩 포함하고 있는 수열이다. 수열 P를 길이가 N인 배열 A에 적용하면 길이가 N인 배열 B가 된다. 적용하는 방법은 B[P[i]] = A[i]이다.

배열 A가 주어졌을 때, 수열 P를 적용한 결과가 비내림차순이 되는 수열을 찾는 프로그램을 작성하시오. 비내림차순이란, 각각의 원소가 바로 앞에 있는 원소보다 크거나 같을 경우를 말한다. 만약 그러한 수열이 여러개라면 사전순으로 앞서는 것을 출력한다.

입력

첫째 줄에 배열 A의 크기 N이 주어진다. 둘째 줄에는 배열 A의 원소가 0번부터 차례대로 주어진다. N은 50보다 작거나 같은 자연수이고, 배열의 원소는 1,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 비내림차순으로 만드는 수열 P를 출력한다.

 

풀이

문제를 이해하는 것부터 굉장히 어려웠다.

주어진 예시를 통해 문제를 차근차근 이해해보도록 하자.

A의 원소로 {2, 3, 1}이 주어졌고, P를 통해 비내림차순 수열인 B를 만들어주어야 한다.

A[0] = 2 = B[1] = B[P[0]] → P[0] = 1

A[1] = 3 = B[2] = B[P[1]] → P[1] = 2

A[2] = 1 = B[0] = B[P[2]] → P[2] = 0

가 되는 것이다.

B가 비내림차순이기 때문에 1인 원소의 index가 0이 되고, 2인 원소의 index가 1이 되고, 3인 원소의 index가 2가 되는 것이다.

 

배열의 원소와 index를 같이 pair로 같이 저장해준다.

배열의 원소의 크기에 따라 정렬을 진행해주고, 작은 원소부터 새로운 index를 부여해주도록 한다.

코드

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int n;
 
int main() {
    cin >> n;
 
    vector<pair<int, int>>val_index;
    for (int i = 0; i < n; i++) {
        int val;
        cin >> val;
        val_index.push_back(make_pair(val, i));
    }
 
    sort(val_index.begin(), val_index.end());
 
    vector<int> tmp(n);
    for (int i = 0; i < n; i++) {
        tmp[val_index[i].second] = i;
    }
 
    for (auto iter : tmp) {
        cout << iter << ' ';
    }
}

 

메모

make_pair를 이용하여 pair를 만들어볼 수 있었다. pair에 아직 익숙하진 않지만 점점 여러 메소드들의 활용법을 익혀나가는 것 같아 기분이 좋다.