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

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 선택 정렬로 배열 A를 오름차순 정렬할 경우 정렬 과정에서 배열 A가 배열 B와 같은 경우가 발생하는지 확인해 보자. 초기 상태 배열 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, B의 크기 N(5 ≤ N ≤ 500,000)이 주어진다.

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

다음 줄에 서로 다른 배열 B의 원소 B1, B2, ..., BN이 주어진다. (1 ≤ Bi ≤ 109)

 

출력

선택 정렬로 배열 A를 오름차순 정렬하는 과정에서 배열 A가 배열 B와 같은 경우가 발생하면 1, 아니면 0을 출력한다.

 

---

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

---

풀이

#include <iostream>
#include <algorithm>
#include <unordered_map>
#include <vector>
using namespace std;

vector<int> A(500001);
vector<int> B(500001);
unordered_map <int, int> m;

// 선택 정렬
int selection_sort(int N) {
	if (A == B) return 1;

	vector<int> arr(A);
	sort(arr.begin(), arr.begin() + N);

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		int big = m[arr[i]]; // 가장 큰 수의 위치

		// 다르면 교환이 발생
		if (i != big) {
			int tmp = A[i];
			A[i] = A[big];
			A[big] = tmp;

			m[A[big]] = big;

			// 배열 A가 배열 B와 같은 경우
			if (A == B) return 1;
		}
	}

	return 0;
}

int main() {
	int N; //  배열의 크기 N
	scanf("%d", &N);

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

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

	cout << selection_sort(N) << endl;

	return 0;
}

처음에는 map을 사용해서 https://nation130.tistory.com/205 와 비슷하게 풀었으나 시간초과가 발생했다. 찾아보니 map은 삽입/삭제 연산에서 시간이 오래 걸리기에 차라리 입력을 한 번에 받고 한 번에 이를 정렬하는 것이 빠르다는 사실을 알 수 있었다. 그래서 이번에는 unordered_map으로 문제를 풀었다.

 

// 선택 정렬
int selection_sort(int N) {
	if (A == B) return 1;

	vector<int> arr(A);
	sort(arr.begin(), arr.begin() + N);

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		int big = m[arr[i]]; // 가장 큰 수의 위치

		// 다르면 교환이 발생
		if (i != big) {
			int tmp = A[i];
			A[i] = A[big];
			A[big] = tmp;

			m[A[big]] = big;

			// 배열 A가 배열 B와 같은 경우
			if (A == B) return 1;
		}
	}

	return 0;
}

arr는 A를 정렬한 결과물로 여기서 가장 큰 수를 바로 찾을 수 있다. arr[i]가 가장 큰 수인데, 이 수를 key로 사용해 m에서 해당 숫자가 A에서 어디에 있는지를 찾아 big에 저장했다.

이 big이 i와 같지 않으면 해당 수는 제자리에 있는 것이 아니기에 올바른 자리로 교환하도록 했다. 이렇게 교환이 발생할 때마다 A와 B를 비교해서 서로 같은 지 확인했다.