[23881] 알고리즘 수업 - 선택 정렬 1 (JAVA)

# 문제 설명

오늘도 서준이는 선택 정렬 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 선택 정렬로 배열 A를 오름차순 정렬할 경우 K 번째 교환되는 수를 구해서 우리 서준이를 도와주자.

크기가 N인 배열에 대한 선택 정렬 의사 코드는 다음과 같다.

selection_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
    for last <- N downto 2 {
        A[1..last]중 가장 큰 수 A[i]를 찾는다
        if (last != i) then A[last] <-> A[i]  # last와 i가 서로 다르면 A[last]와 A[i]를 교환
    }
}

---

입력

첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N)가 주어진다.

다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)

---

출력

K 번째 교환되는 두 개의 수를 작은 수부터 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.

 

# 정답 코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken()); //배열의 크기
        int K = Integer.parseInt(st.nextToken()); //교환 횟수

        int[] A = new int[N];
        st = new StringTokenizer(br.readLine());
        for (int n = 0; n < N; n++) {
            A[n] = Integer.parseInt(st.nextToken());
        }

        int cnt = 0;

        for (int i = A.length-1; i > 0; i--) {
            int maxIdx = i;

            for (int j = 0; j < i; j++) {
                if (A[maxIdx] < A[j]) {
                    maxIdx = j;
                }
            }

            if (maxIdx != i) {
                cnt++;

                if (cnt == K) {
                    System.out.print(A[i] + " " + A[maxIdx]);
                    return;
                }

                swap(A, maxIdx, i);
            }
        }

        System.out.print("-1");
    }

    public static void swap(int[] arr, int a, int b) {
        int num = arr[a];
        arr[a] = arr[b];
        arr[b] = num;
    }
}

 

문제 예시를 보면 정렬 범위 내에서의 최댓값을 찾아 정렬 범위의 맨 뒤와 swap하여 오름차순 정렬을 한다.

 

i는 정렬 범위를 조정하는데, 맨 뒤인 A의 길이-1부터 1까지 1씩 감소한다.

maxIdx를 만들어 최댓값 인덱스를 저장한다.

j로 0부터 i까지 돌며 정렬 범위 내의 최댓값을 찾는다.

 

만약 최댓값이 i가 아니면 A[i]와 A[maxIdx]를 swap해주면 되므로 cnt를 1 증가한다.

증가한 cnt가 K와 같으면, 숫자가 작은 순서대로 출력해주면 된다.

 

교환 횟수 cnt가 K보다 작으면 -1을 출력한다.