백준 1015 - 수열 정렬

문제

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

1015번: 수열 정렬

 

 

풀이

아래 그림에서 P 배열을 구하는 문제이다. A 배열을 정렬해 B 배열을 만들고, A와 B를 이용해 P배열을 구하면 된다. 

B 배열이 정렬돼 있으므로 B 배열에 있는 값과 A 배열에 있는 값을 비교해 P 배열에 값을 넣을 수 있다. B 배열에 있는 값을 A 배열에서 찾은 후, 찾은 A 배열의 인덱스를 그대로 몇 번째로 찾았는지 P 배열에 저장하면 된다.

 

 

코드

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    static int N;
    static int[] A, B, P;
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) {
        input();
        solve();
        System.out.println(sb.toString());
    }

    private static void solve() {
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (A[j] == B[i] && P[j] == -1) {
                    P[j] = i;
                    break;
                }
            }
        }

        for (int i : P) {
            sb.append(i + " ");
        }
    }

    private static void input() {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        A = new int[N];
        B = new int[N];
        P = new int[N];

        for (int i = 0; i < N; i++) {
            A[i] = sc.nextInt();
            B[i] = A[i];
        }

        Arrays.fill(P, -1);
        Arrays.sort(B);
    }
}