[Baekjoon/C++] 23968번 - 알고리즘 수업 - 버블 정렬 1

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 버블 정렬로 배열 A를 오름차순 정렬할 경우 K 번째 교환되는 수를 구해서 우리 서준이를 도와주자.

크기가 N인 배열에 대한 버블 정렬 의사 코드는 다음과 같다.

bubble_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
    for last <- N downto 2
        for i <- 1 to last - 1
            if (A[i] > A[i + 1]) then A[i] <-> A[i + 1]  # 원소 교환
}

 

입력

첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N2)가 주어진다.

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

 

출력

K 번째 교환되는 두 개의 수를 작은 수부터 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.

 

---

예제 입력	예제 출력
6 10 4 6 5 1 3 2	2 4
6 12 4 6 5 1 3 2	-1

---

풀이

#include <iostream>
using namespace std;

int bubble_sort(int A[]);

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

// 버블 정렬 
int bubble_sort(int A[]) {
	int count = 0; // 교환 횟수

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		for (int j = 0; j < i; j++) {
			// 원소 교환
			if (A[j] > A[j + 1]) {
				int tmp = A[j + 1];
				A[j + 1] = A[j];
				A[j] = tmp;
				
				count++;
			}

			// K 번째 교환되는 수의 인덱스
			if (count == K) return j;
		}
	}

	return -1;
}

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

	int* A = new int[N];

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

	int idx = bubble_sort(A);

	// 출력
	if (idx == -1)
		printf("%d\n", idx);
	else
		printf("%d %d\n", A[idx], A[idx + 1]);

	delete[] A;

	return 0;
}

 

// 버블 정렬 
int bubble_sort(int A[]) {
	int count = 0; // 교환 횟수

	//A[1..N]을 오름차순 정렬한다.
	for (int i = N - 1; i > 0; i--) {
		for (int j = 0; j < i; j++) {
			// 원소 교환
			if (A[j] > A[j + 1]) {
				int tmp = A[j + 1];
				A[j + 1] = A[j];
				A[j] = tmp;
				
				count++;
			}

			// K 번째 교환되는 수의 인덱스
			if (count == K) return j;
		}
	}

	return -1;
}

위 코드로 구현한 버블 정렬은 오름차순 정렬이다. 인접한 두 숫자의 크기를 비교하고, 더 큰 수를 뒤로 보내게 된다. 이렇게 원소 교환이 발생하면 count의 값을 1씩 증가해서 해당 교환이 몇 번째 교환인지를 셌다. 그러다 count가 K과 같게 되면 현재 교환이 일어난 인덱스인 j를 return하게 된다.

만약 모든 교환이 끝났는데도 count가 K와 같지 않다면 -1을 return한다.

 

	int idx = bubble_sort(A);

	// 출력
	if (idx == -1)
		printf("%d\n", idx);
	else
		printf("%d %d\n", A[idx], A[idx + 1]);

위의 코드를 통해 받은 값이 -1이면 -1을 출력하고, -1이 아니면 교환이 발생한 두 수를 작은 것부터 출력한다.