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

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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

N이 매우 커서 시간 초과를 고민하고 있는 우리 서준이를 도와주자.

크기가 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 ≤ 500,000), 교환 횟수 K(1 ≤ K ≤ N)가 주어진다.

다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)

 

출력

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

 

---

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

---

풀이

#include <iostream>
#include <map>
using namespace std;

int selection_sort(int N, int K);

int A[500001];
map <int, int> m;

// 선택 정렬
int selection_sort(int N, int K) {
	int count = 0;
	int idx = N - 1;

	map <int, int>::reverse_iterator it = m.rbegin();

	for (it = m.rbegin(); it != m.rend(); ++it) {
		int big = it->second; // A[]중 가장 큰 수

		// last와 big이 서로 다르면 교환이 발생한다.
		if (big != idx) {
			int tmp = A[idx];
			A[idx] = it->first;
			A[big] = tmp;

			// 각 숫자의 현재 인덱스 위치
			it->second = idx;
			map<int, int>::iterator find = m.find(tmp);
			find->second = big;

			count++;
		}

		// K 번째 교환
		if (count == K) {
			for (int i = 0; i < N; i++)
				printf("%d ", A[i]);
			printf("\n");

			return 0;
		}

		idx--;
	}

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

	return 0;
}

int main() {
	int N, K; //  배열 A의 크기 N, 교환 횟수 K
	scanf("%d %d", &N, &K);

	for (int i = 0; i < N; i++) {
		scanf("%d", &A[i]);
		m.insert(map<int, int>::value_type(A[i], i));
	}

	selection_sort(N, K);

	return 0;
}

https://nation130.tistory.com/204

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

// K 번째 교환
if (count == K) {
	for (int i = 0; i < N; i++)
		printf("%d ", A[i]);
	printf("\n");
    
	return 0;
}

K 번째 교환이 발생했을 때 값을 출력하는 위의 코드를 뺀 나머지는 어제 올린 게시글의 문제 풀이와 같다.