[Baekjoon/C++] 24052번 - 알고리즘 수업 - 삽입 정렬 2

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 삽입 정렬로 배열 A를 오름차순 정렬할 경우 배열 A의 원소가 K 번 변경된 직후의 배열 A를 출력해 보자.

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

insertion_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
    for i <- 2 to N {
        loc = i - 1;
        newItem = A[i];

        # 이 지점에서 A[1..i-1]은 이미 정렬되어 있는 상태
        while (1 <= loc and newItem < A[loc]) {
            A[loc + 1] <- A[loc];
            loc--;
        }
        if (loc + 1 != i) then A[loc + 1] = newItem;
    }
}

 

입력

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

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

 

출력

K 번 변경이 발생한 직후의 배열 A를 한 줄에 출력한다. 변경 횟수가 K 보다 작으면 -1을 출력한다.

 

---

예제 입력	예제 출력
5 7 4 5 1 3 2	1 3 4 5 5
5 11 4 5 1 3 2	-1

---

풀이

#include <iostream>
using namespace std;

int insertion_sort(int N, int K, int A[]);

// 삽입 정렬
int insertion_sort(int N, int K, int A[]) {
    int count = 0; // 저장 횟수

    //A[]을 오름차순 정렬한다.
    for (int i = 1; i < N; i++) {
        int loc;
        int newItem = A[i];

        // 이 지점에서 A[1..i - 1]은 이미 정렬되어 있는 상태
        for (loc = i - 1; loc >= 0 && newItem < A[loc]; loc--) {
            A[loc + 1] = A[loc];
            count++;

            // K 번째 저장
            if (count == K) return 1;
        }

        if (loc + 1 != i) {
            A[loc + 1] = newItem;
            count++;

            // K 번째 저장
            if (count == K) return 1;
        }
    }

    return -1;
}

int main() {
	int N, K; //  배열의 크기 N, 저장 횟수 K
	scanf("%d %d", &N, &K);

    int* A = new int[N];

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

    // 출력
    if (insertion_sort(N, K, A) == -1)
        printf("-1\n");
    else {
        for (int i = 0; i < N; i++)
            printf("%d ", A[i]);
        printf("\n");
    }

	return 0;
}

배열에서 순서대로 숫자를 선택해서 newItem에 저장하고 해당 숫자가 맞는 자리에 들어갈 수 있게 숫자들을 뒤로 한 칸씩 옮겨준다. 이때 숫자를 옮기면서 배열에 변경이 발생하고, newItem의 숫자가 만들어든 자리에 들어갈 때도 변경이 발생하므로 count를 1씩 증가하는 부분과 변경 발생 횟수를 K와 비교하는 부분이 for문 안에만 2번 넣어주었다. 그리고 K번째 변경이 발생하면 1을, 모든 정렬이 끝났는데도 K번째 변경이 없다면 -1을 return하게끔 했다.

이렇게 return 받은 숫자로 K번째 변경이 발생한 경우와 그렇지 않은 경우를 구별해서 출력해줬다.