[백준] 24060번 알고리즘 수업 - 병합 정렬 1 (C++) **

문제링크

www.boj.kr/24060 

 

소스코드 (C++)

#include <iostream>
#include <stdlib.h>

using namespace std;

int num = 0;
int length;
int K;
int flag = 0;
void merge(int arr[], int l, int m, int r) {
    int i, j, k;
    int n1 = m - l + 1;
    int n2 = r - m;

    int* L = new int[n1];
    int* R = new int[n2];

    for (i = 0; i < n1; i++)
        L[i] = arr[l + i];
    for (j = 0; j < n2; j++)
        R[j] = arr[m + 1 + j];

    i = 0;
    j = 0;
    k = l;
    while (i < n1 && j < n2) {
        if (L[i] <= R[j]) {
            arr[k] = L[i];
            num++;
            if (num == K) {
                printf("%d", arr[k]);
                flag = 1;
            }
            i++;
        }
        else {
            arr[k] = R[j];
            num++;
            if (num == K) {
                printf("%d", arr[k]);
                flag = 1;
            }
            j++;
        }
        k++;
    }

    while (i < n1) {
        arr[k] = L[i];
        num++;
        if (num == K) {
            printf("%d", arr[k]);
            flag = 1;
        }
        i++;
        k++;
    }

    while (j < n2) {
        arr[k] = R[j];
        num++;
        if (num == K) {
            printf("%d", arr[k]);
            flag = 1;
        }
        j++;
        k++;
    }
}

void mergeSort(int arr[], int l, int r) {
    if (l < r) {
        int m = l + (r - l) / 2;

        mergeSort(arr, l, m);
        mergeSort(arr, m + 1, r);

        merge(arr, l, m, r);
    }
}

int main() {
    cin >> length;
    cin >> K;

    int* arr = new int[length];

    for (int i = 0; i < length; i++) {
        cin >> arr[i];
    }
      
    mergeSort(arr, 0, length - 1);

    if (flag == 0) {
        printf("-1");
    }
    return 0;
}

 

 

설명

병합정렬을 이용해서 푼다.