백준 알고리즘 수업 - 선택 정렬 2

KIMYEONGJUN·2025년 2월 8일
post-thumbnail

문제

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

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

K 번 교환이 발생한 직후의 배열 A를 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.

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

입력 처리: BufferedReader와 StringTokenizer를 사용하여 입력을 받는다. 
첫 번째 줄에서 배열의 크기 N과 교환 횟수 K를 읽고, 두 번째 줄에서 배열의 요소들을 읽는다.
선택 정렬 구현: 배열을 선택 정렬 방식으로 정렬해준다.
각 단계에서 최대값의 인덱스를 찾고, 현재 위치와 다르면 두 수를 교환합니다. 교환 횟수를 카운트하고, K번째 교환이 발생했을 때 배열을 출력한다.
결과 출력: 교환 횟수가 K에 도달하면 해당 배열을 출력하고, 그렇지 않으면 -1을 출력한다.

코드로 구현

package baekjoon.baekjoon_26;

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

// 백준 23882번 문제
public class Main926 {
    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 i = 0; i < N; i++) {
            A[i] = Integer.parseInt(st.nextToken());
        }

        int swapCount = 0;
        boolean found = false;

        for(int last = N; last >= 2; last--) {
            int maxIndex = 0;
            for(int i = 1; i < last; i++) {
                if (A[i] > A[maxIndex]) {
                    maxIndex = i;
                }
            }

            if(last - 1 != maxIndex) {
                int temp = A[last - 1];
                A[last - 1] = A[maxIndex];
                A[maxIndex] = temp;
                swapCount++;

                if(swapCount == K) {
                    found = true;
                    break;
                }
            }
        }

        if(found) {
            for(int num : A) {
                System.out.print(num + " ");
            }
        } else {
            System.out.println(-1);
        }

        br.close();
    }
}

마무리

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

profile
Junior backend developer

0개의 댓글