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

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

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

크기가 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 번째 저장 되는 수를 출력한다. 저장 횟수가 K 보다 작으면 -1을 출력한다.

 

---

예제 입력	예제 출력
5 7 4 5 1 3 2	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 A[loc + 1];
        }

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

            // K 번째 저장
            if (count == K) return A[loc + 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]);

    // 출력
	cout << insertion_sort(N, K, A) << endl;

	return 0;
}

count를 0으로 초기화하고 저장이 발생할 때마다 1씩 증가시켜서 K번째에 저장한 숫자를 찾았다. 삽입 정렬에서는 저장하는 코드가 2번 나오므로 count가 K와 같은지 확인하는 if문도 2번 넣었다.