[Baekjoon/C++] 23881번 - 알고리즘 수업 - 선택 정렬 1

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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을 출력한다.

 

---

예제 입력	예제 출력
5 2 3 1 2 5 4	2 3
5 4 3 1 2 5 4	-1

---

풀이

#include <iostream>
using namespace std;

int N, K; // 배열 A의 크기 N, 교환 횟수 K

// 선택 정렬
int selection_sort(int A[]) {
	int count = 0; // 교환 횟수

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		int big = 0;

		// A[]중 가장 큰 수 A[]를 찾는다
		for (int j = 0; j <= i; j++)
			if (A[j] > A[big]) big = j;

		// last와 big이 서로 다르면 A[last]와 A[big]을 교환
		if (i != big) {
			int tmp = A[i];
			A[i] = A[big];
			A[big] = tmp;

			count++;
		}
		
		// K 번째 교환되는 두 개의 수
		if (count == K) {
			printf("%d %d\n", A[big], A[i]);

			return 0;
		}
	}

	return -1;
}

int main() {
	cin >> N >> K;

	int* A = new int[N];

	for (int i = 0; i < N; i++)
		scanf("%d", &A[i]);

	int num = selection_sort(A);

	// 출력
	if (num == -1) printf("-1\n");

	delete[] A;

	return 0;
}

 

// 선택 정렬
int selection_sort(int A[]) {
	int count = 0; // 교환 횟수

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		int big = 0;

		// A[]중 가장 큰 수 A[]를 찾는다
		for (int j = 0; j <= i; j++)
			if (A[j] > A[big]) big = j;

		// last와 big이 서로 다르면 A[last]와 A[big]을 교환
		if (i != big) {
			int tmp = A[i];
			A[i] = A[big];
			A[big] = tmp;

			count++;
		}
		
		// K 번째 교환되는 두 개의 수
		if (count == K) {
			printf("%d %d\n", A[big], A[i]);

			return 0;
		}
	}

	return -1;
}

배열 A에서 가장 큰 수의 위치를 big에 저장한 뒤 big이 i와 같은 지 확인한다. i는 아직 정렬되지 않은 수 중 가장 뒤에 있는 수의 위치를 나타내므로 big이 i와 같지 않으면 해당 숫자는 정렬되지 않은 것이다. 그러므로 A[i]와 A[big]의 원소를 서로 교환하고, 교환 횟수 count를 1 증가시켰다.

만약 해당 교환이 K번째 교환이라면 교환한 두 원소를 출력하고 0을 return해서 함수를 끝냈다.

만약 모든 교환이 끝났는데도 K번째 교환이 없다면 -1을 return했다.