Baekjoon 23882 알고리즘 수업 - 선택 정렬 2 JAVA

 

23882번: 알고리즘 수업 - 선택 정렬 2

문제

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

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 선택 정렬로 배열 A를 오름차순 정렬할 경우 K 번 교환이 발생한 직후의 배열 A를 출력해 보자.

크기가 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 번 교환이 발생한 직후의 배열 A를 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.

풀이

2023.03.26 - [Algorithm/Baekjoon] - Baekjoon 23881 알고리즘 수업 - 선택 정렬 1 JAVA

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

위의 문제와 마찬가지로 교환되는 횟수를 카운팅 하는 문제이다.

다만 다른 점이 있다면 만족하는 교환 횟수를 찾았을 때 현재 상태를 출력하는 것이다.

코드

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());
		StringBuilder sb = new StringBuilder();
		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 max=0, idx=0;
			for(int j=0;j<i;j++) {
				if(max<A[j]) {
					max=A[j];
					idx=j;
				}
			}
			if(A[i]<A[idx]) {
				int temp = A[i];
				A[i]=A[idx];
				A[idx]=temp;
				cnt++;
			}			
			if(cnt==k)
				break;
		}
		if(cnt<k)
			System.out.println(-1);
		else {
			for(int i=0;i<n;i++)
				sb.append(A[i]+" ");
			System.out.println(sb.toString());
		}
	}
}