백준 | 정렬 섹션 - 23881번. 선택 정렬 1

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 ≤ 109)

출력

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

내 코드 | 첫 시도

import java.util.ArrayList;
import java.util.Scanner;

/*백준. 23881번. 선택 정렬 1 | 다시 풀어야 함 ㅠ
*/
public class Main {
    //솔루션 함수
    public ArrayList<Integer> solution(int n, int k, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();
        int cnt = 0;
        for(int i = 0; i<n-1; i++) {
            int idx = i;
            for(int j=i+1; j<n; j++) {
                if(arr[j]<arr[idx]) idx = j;
            }
            //swap 교환
            cnt++;
            if(cnt == k) {
                answer.add(arr[idx]);
                answer.add(arr[i]);
            }
            int tmp = arr[i];
            arr[i] = arr[idx];
            arr[idx] = tmp;
        }

        if(cnt < k) answer.add(-1);

        return answer;
    }    
    //실행 메인
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Main T = new Main();
        Scanner kb = new Scanner(System.in);

        int n = kb.nextInt();
        int k = kb.nextInt();

        int[] arr = new int[n];
        for(int i = 0; i<n; i++) arr[i] = kb.nextInt();

        for(int x : T.solution(n, k, arr)) System.out.print(x+ " ");    
    }
}

내 코드 | 두 번째 시도

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

/*백준. 23881번. 선택 정렬 1
*/
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 [] arr = new int[N+1];

        st=new StringTokenizer(br.readLine());
        for (int i=1;i<=N;i++){
            arr[i]=Integer.parseInt(st.nextToken());
        }

        int cnt = 0;

        for (int last=N; last>1; last--){
          int max = arr[0];
          int maxIdx = 0;
          for (int i=1;i<=last;i++){
            if(max < arr[i]){
              max = arr[i];
              maxIdx = i;
            }
          }
          if(last!=maxIdx){
            int temp = arr[last];
            arr[last] = arr[maxIdx];
            arr[maxIdx] = temp;
            cnt ++;

            if(cnt == K){
              System.out.println(arr[maxIdx]+" "+arr[last]);
              System.exit(0);
            }
          }
        }

        if(cnt!=K){
          System.out.println(-1);
        }
      }

}