[Algorithm] Merge Sort (c++)

Jihoon Kim·2024년 2월 5일
0
#include <iostream>

using namespace std;

// 두 개의 정렬된 배열을 병합하는 함수
void merge(int arr[], int left, int mid, int right) {
    int n1 = mid - left + 1;
    int n2 = right - mid;

    // 임시 배열들을 생성하고 데이터를 복사
    int leftArr[n1], rightArr[n2];

    for (int i = 0; i < n1; i++)
        leftArr[i] = arr[left + i];
    for (int j = 0; j < n2; j++)
        rightArr[j] = arr[mid + 1 + j];

    // 두 개의 임시 배열을 합병
    int i = 0, j = 0, k = left;
    while (i < n1 && j < n2) {
        if (leftArr[i] <= rightArr[j]) {
            arr[k] = leftArr[i];
            i++;
        } else {
            arr[k] = rightArr[j];
            j++;
        }
        k++;
    }

    // 남은 요소들 복사
    while (i < n1) {
        arr[k] = leftArr[i];
        i++;
        k++;
    }

    while (j < n2) {
        arr[k] = rightArr[j];
        j++;
        k++;
    }
}

// 병합 정렬 함수
void mergeSort(int arr[], int left, int right) {
    if (left < right) {
        // 중간 지점 계산
        int mid = left + (right - left) / 2;

        // 왼쪽과 오른쪽 부분 배열을 정렬
        mergeSort(arr, left, mid);
        mergeSort(arr, mid + 1, right);

        // 정렬된 부분 배열을 병합
        merge(arr, left, mid, right);
    }
}

int main() {
    // 테스트 배열
    int arr[] = {12, 11, 13, 5, 6, 7};
    int n = sizeof(arr) / sizeof(arr[0]);

    // 배열을 출력
    cout << "정렬 전 배열: ";
    for (int i = 0; i < n; i++)
        cout << arr[i] << " ";
    cout << endl;

    // 병합 정렬 수행
    mergeSort(arr, 0, n - 1);

    // 정렬된 배열 출력
    cout << "정렬 후 배열: ";
    for (int i = 0; i < n; i++)
        cout << arr[i] << " ";
    cout << endl;

    return 0;
}
profile
AI/ML Research Engineer

0개의 댓글