AcWing 787. 归并排序(递归, 模板)

题目描述

原题链接

分析

归并排序的基本思想是, 将两个有序表合并成一个有序表

实现

#include <cstdio>
#include <iostream>
using namespace std;
const int N = 1e5 + 9;
int n;
int a[N];
int temp[N]; // 辅助空间
void mergeSort(int l, int r) // [l,r]将要合并的区间
{
    if(l >= r) return;
    int mid = ( l + r ) / 2;
    mergeSort(l, mid);
    mergeSort(mid+1,r);
    
    // 有序表的合并
    int i = l, j = mid + 1, cnt = 0;
    while(i <= mid && j <= r)
    {
        if(a[i] < a[j]) temp[cnt++] = a[i++];
        else temp[cnt++] = a[j++];
    }
    while(i<=mid) temp[cnt++] = a[i++];
    while(j<=r) temp[cnt++] = a[j++];
    
    // 将结果复制回原数组
    for(int i=l, j=0; i<=r; i++, j++) a[i] = temp[j];
}
int main()
{
    cin >> n;
    for(int i=0; i<n; i++) cin >> a[i];
    mergeSort(0,n-1);
    for(int i=0; i<n; i++) cout << a[i] << " ";
    return 0;
}