백준 알고리즘 수업 - 삽입 정렬 1

문제

내가 생각했을때 문제에서 원하는부분

첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 저장 횟수 K(1 ≤ K ≤ N2)가 주어진다.
다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 10^9)

K 번째 저장 되는 수를 출력한다. 저장 횟수가 K 보다 작으면 -1을 출력한다.

내가 이 문제를 보고 생각해본 부분

입력 처리: BufferedReader를 사용하여 입력을 받는다.
삽입 정렬: 주어진 배열을 오름차순으로 정렬하며, 저장 횟수를 카운트해준다.
K번째 저장 확인: 저장 횟수가 K에 도달할 때마다 해당 값을 기록한다.
결과 출력: K번째 저장되는 수를 출력하며, 저장 횟수가 K보다 적으면 -1을 출력한다.

코드로 구현

package baekjoon.baekjoon_26;

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

// 백준 24051번 문제
public class Main927 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        // 입력 받기
        String[] firstLine = br.readLine().split(" ");
        int N = Integer.parseInt(firstLine[0]);
        int K = Integer.parseInt(firstLine[1]);

        int[] A = new int[N];
        String[] secondLine = br.readLine().split(" ");
        for(int i = 0; i < N; i++) {
            A[i] = Integer.parseInt(secondLine[i]);
        }

        int saveCount = 0; // 저장 횟수
        int KthSaveValue = -1; // K 번째 저장되는 수

        // 삽입 정렬
        for(int i = 1; i < N; i++) {
            int newItem = A[i];
            int loc = i - 1;

            // A[0..i-1]은 이미 정렬되어 있음
            while(loc >= 0 && A[loc] > newItem) {
                A[loc + 1] = A[loc];
                loc--;
                saveCount++;

                // K 번째 저장을 찾음
                if(saveCount == K) {
                    KthSaveValue = A[loc + 1];
                }
            }

            if(loc + 1 != i) {
                A[loc + 1] = newItem;
                saveCount++;

                // K 번째 저장을 찾음
                if(saveCount == K) {
                    KthSaveValue = newItem;
                }
            }
        }

        // 결과 출력
        System.out.println(KthSaveValue);
        br.close();
    }
}

마무리

코드와 설명이 부족할수 있습니다. 코드를 보시고 문제가 있거나 코드 개선이 필요한 부분이 있다면 댓글로 말해주시면 감사한 마음으로 참고해 코드를 수정 하겠습니다.