[ BOJ ] 23968 : 알고리즘 수업 - 버블 정렬 1 ( BRONZE 1 ) / C, Python

문제

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

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

풀이 과정

문제에서 버블 정렬 수도 코드를 제공해주기에 저 코드대로 구현하면 된다.

수를 교환할 때마다 카운팅해서, K 번째 교환되는 두 개의 수를 출력한다.

C

#include <stdio.h>

int arr[10001];
int main(void) {
    int n, k;
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; i++) scanf("%d", &arr[i+1]);

    int count = 0;
    for (int last = n; last >= 2; last--) {
        for (int i = 1; i < last; i++) {
            if (arr[i] > arr[i+1]) {
                int tmp = arr[i];
                arr[i] = arr[i+1];
                arr[i+1] = tmp;

                count += 1;
                if (count == k) {
                    printf("%d %d", arr[i], arr[i+1]);
                    return 0;
                }
            }
        }
    }

    printf("-1");
    return 0;
}

Python

import sys
input = sys.stdin.readline

n, k = map(int, input().rstrip().split())
arr = list(map(int, input().rstrip().split()))
cnt = 0
res = []
for i in range(n, 1, -1):
    for j in range(i - 1):
        if arr[j] > arr[j + 1]:
            cnt += 1

            arr[j:j + 2] = arr[j:j + 2][::-1]
            if cnt == k: res = [arr[j], arr[j+1]]

if len(res) == 0: print(-1)
else: print(' '.join(map(str, res)))