본문 바로가기

BOJ/Node.js

[BOJ / Node.js] 23968. 알고리즘 수업 - 버블 정렬 1

728x90
 

23968번: 알고리즘 수업 - 버블 정렬 1

첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N2)가 주어진다. 다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)

www.acmicpc.net

문제

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

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

 

Solve

  • 버블 정렬을 구현하고, 배열 안의 원소가 정렬이 되는 과정에서 swap할 때마다 count를 1씩 증가시켜 k와 같아질 경우 swap되는 값을 출력하고 반복문을 break하였습니다.

Code

const input = require("fs").readFileSync("/dev/stdin").toString().trim().split("\n");

const [n, k] = input[0].split(' ').map(el => +el);
const arr = input[1].split(' ').map(el=>+el);
let count = 0;
for(let i = n ; i > 0 ; i--) {
    for(let j = 0 ; j < i-1 ; j++) {
        if(arr[j] > arr[j+1]) {
            count++;
            [arr[j], arr[j+1]] = [arr[j+1], arr[j]];
        }
        if(count === k) {
            console.log(arr[j], arr[j+1]);
            break;
        }
    }
    if(count === k) {
        break;
    }
}

if(count < k) console.log(-1)