归并排序

归并排序

归并排序代码

#include <iostream>
#include <vector>
using namespace std;
void merge(vector<int>& nums, int L, int mid, int R) 
{
    vector<int> ans(R - L + 1); // 开辟一个中间数组 大小为 R-L+1
    int p = L, q = mid + 1;
    int i = 0;
    // 将mid 左右两个子数组 排序 ，边界是左边p小于等于 mid， 右边小于等于R
    while (p <= mid && q <= R) 
        ans[i++] = nums[p] <= nums[q] ? nums[p++] : nums[q++];
    // 判断左边是否没有遍历完， 如果没遍历完会逐个遍历追加到 ans 中
    while (p <= mid) 
        ans[i++] = nums[p++];
    // 判断右边是否没有遍历完， 如果没遍历完会逐个遍历追加到 ans 中
    while (q <= R) 
        ans[i++] = nums[q++];
    // 将ans中的元素 逐个追加到nums中（注意以L开始 到 R结束）
    for (int i = 0; i < R-L+1; ++i) 
        nums[L + i] = ans[i];
}
void process(vector<int>& nums, int L, int R) {
    if (L == R) return;
    int mid = ((L + R) >> 1);  // 将数组评分  或者 L + (（L - R） >> 1);
    process(nums, L, mid);
    process(nums, mid + 1, R);
    merge(nums, L, mid, R);
}
void mergeSort(vector<int>& nums) {
    if (nums.size() < 2) return; //当数组元素数量小于等于1时不需要排序
    process(nums, 0, nums.size() - 1);
}
int main() {
    // 输入数据处理
    int n;
    cin >> n;
    vector<int> nums(n);
    for (int i = 0; i < n; ++i) cin >> nums[i];
    // 归并排序
    mergeSort(nums);
    // 排序后打印
    for (int i = 0; i < n; ++i) {
        if (i != 0) cout << ' ';
        cout << nums[i];
    }
    return 0;
}