[1015] 수열 정렬

🔗 문제 링크

https://www.acmicpc.net/problem/1015

🔍 문제 설명

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

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

⚠️ 제한사항

• N은 50보다 작거나 같은 자연수이고, 배열의 원소는 1,000보다 작거나 같은 자연수이다.

🗝 풀이 (언어 : Java)

배열간에 P로 인덱스가 복잡하게 얽혀있어서 헷갈려서 풀기 어려웠다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;

class Main {

    private static void solve(int n, int[][] B) {
        Arrays.sort(B, Comparator.comparingInt(o -> o[0]));
        int[] P = new int[n];
        for (int i = 0; i < n; i++)
            P[B[i][1]] = i;
        StringBuilder sb = new StringBuilder();
        for (int num : P)
            sb.append(num).append(' ');
        System.out.print(sb);
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] B = new int[n][2];
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        br.close();
        for (int i = 0; i < n; i++) {
            B[i][0] = Integer.parseInt(st.nextToken());
            B[i][1] = i;
        }
        solve(n, B);
    }
}