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

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 버블 정렬로 배열 A를 오름차순 정렬할 경우 정렬 과정에서 배열 A가 배열 B와 같은 경우가 발생하는지 확인해 보자. 초기 상태 배열 A도 정렬 과정에서 발생 가능한 경우로 생각하자.

크기가 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, B의 크기 N(5 ≤ N ≤ 10,000)이 주어진다.

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

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

 

출력

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

 

---

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

---

풀이

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

int* A, * B;

// 버블 정렬
int bubble_sort(int N) {
	// 정렬 전 배열 비교
	if (equal(A, A + N, B)) return 1;

	// 배열을 오름차순 정렬한다.
	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];
				A[j] = A[j + 1];
				A[j + 1] = tmp;

				// A와 B 비교
				if (A[j] == B[j])
					if (equal(A, A + N, B)) return 1;
			}
		}
	}

	return 0;
}


int main() {
	int N; // 배열의 크기 N
	cin >> N;

	A = new int[N];
	B = new int[N];

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

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

	// 출력
	printf("%d\n", bubble_sort(N));

	delete[] A;
	delete[] B;

	return 0;
}

 

// 버블 정렬
int bubble_sort(int N) {
	// 정렬 전 배열 비교
	if (equal(A, A + N, B)) return 1;

	// 배열을 오름차순 정렬한다.
	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];
				A[j] = A[j + 1];
				A[j + 1] = tmp;

				// A와 B 비교
				if (A[j] == B[j])
					if (equal(A, A + N, B)) return 1;
			}
		}
	}

	return 0;
}

버블 정렬을 하는 중간 중간 배열 2개의 값들을 비교하는 함수이다. 먼저 두 배열을 비교해서 이미 같은지 확인한 뒤 정렬을 시작한다. 서로 이웃한 두 수 중 인덱스가 작은 수가 더 클 경우 교환이 발생한다. 교환이 발생하고 나면 두 배열을 비교해서 서로 같으면 1을 return한다. 두 배열의 비교는 최대한 적게 해야 시간 제한에 걸리지 않기 때문에 교환이 발생할 때만 비교하게끔 했다. 만약 모든 정렬이 끝났는데도 두 배열이 같지 않다면 0을 return하게 된다.