Baekjoon 23881 알고리즘 수업 - 선택 정렬 1 JAVA

23881번: 알고리즘 수업 - 선택 정렬 1

문제

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

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 ≤ 10^9)

출력

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

풀이

주어지는 의사 코드를 이용하여 선택 정렬을 구현한다

위의 선택 정렬은 가장 큰 값부터 정렬하는 형태로 구현이 되어 있다

입력받는 for문 (for(int i=0;i<n;i++)은 기본적으로 입력을 위한 것이며

 

바로 뒤에 등장하는 for(int i=n-1;i>0;i--)는 가장 큰 값을 정하기 위한 배열의 마지막부터 비교하는 반복문이다.

for(int j=0;j<=i;j++) 을 이용하여 주어진 배열중에서 가장 큰 값을 찾아준다.

이렇게 찾은 값을 이용하여 if문에서 배열중 마지막 위치가 현재 찾은 값과 다르다면 교환이 일어나기에 cnt를 증가시킨다.

 

위의 과정을 반복하면서 cnt가 k와 동일해 지는 경우 교환되는 위치의 값을 출력해주고 반복을 종료한다.

 

만약 cnt가 k보다 작다면 -1을 리턴하면서 종료시킨다. 

코드

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

public class Main {
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int n= Integer.parseInt(st.nextToken()), k=Integer.parseInt(st.nextToken()), cnt=0;
		int [] a = new int [n];
		st = new StringTokenizer(br.readLine());
		for(int i=0;i<n;i++)
			a[i]=Integer.parseInt(st.nextToken());
		for(int i=n-1;i>0;i--) {
			int m=0, idx=0;
			for(int j=0;j<=i;j++) {
				if(m<a[j]) {
					m=a[j];
					idx=j;
				}
			}
			if(a[i]!=m) {
				int t=a[i];
				a[i]=a[idx];
				a[idx]=t;
				cnt++;
				if(cnt==k) {
					System.out.println(Math.min(a[i], a[idx])+" "+Math.max(a[i], a[idx]));
					break;
				}
			}
		}
		if(cnt<k)
			System.out.println("-1");
	}
}