力扣面试经典150 88. 合并两个有序数组 归并排序的merge函数

题目链接

规定了时间我首先想到的是归并排序,以时间换空间,题目要求时间复杂度O(m+n)

提示:

复制代码
nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[j] <= 109

而且这个题理论上,是合并,所以得把最后合并的结果放到nums1[]上

下面是代码

java 复制代码
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
   vector<int> temp(nums1.begin(),nums1.begin()+m);
   int p=0;
   int p1=0;
   int p2=0;
   while(p1<m&&p2<n){if(temp[p1]<=nums2[p2])
    nums1[p++]=temp[p1++];else
    nums1[p++]=nums2[p2++];
   }

while(p1<m)nums1[p++]=temp[p1++];
while(p2<n)nums1[p++]=nums2[p2++];


    }
};

然后再复习一下归并排序

这么多年了,发现自己学习算法过于死板

其实归并排序是一棵归并排序二叉树。

那么一涉及到树就很容易理解,这里头有递归,所以归并排序是由递归和merge函数组合来实现的,那个大的递归函数就是

java 复制代码
void mergeSort(vector<int>& arr, int l, int r){
if(l>=r) return;
int mid=(l+r)/2;
mergesort(arr,l,mid);
mergesort(arr,mid,r);
merge(arr,l,mid,r);
}

下面是完整的归并排序的代码

java 复制代码
#include <iostream>
#include <vector>
using namespace std;

// 合并两个有序区间 [l, mid] 和 [mid+1, r]
void merge(vector<int>& arr, int l, int mid, int r) {
    vector<int> temp(r - l + 1);
    int i = l;
    int j = mid + 1;
    int k = 0;

    while (i <= mid && j <= r) {
        if (arr[i] <= arr[j]) {
            temp[k++] = arr[i++];
        } else {
            temp[k++] = arr[j++];
        }
    }

    while (i <= mid) temp[k++] = arr[i++];
    while (j <= r) temp[k++] = arr[j++];

    // 复制回原数组
    for (int p = 0; p < temp.size(); ++p) {
        arr[l + p] = temp[p];
    }
}

// 归并排序递归
void mergeSort(vector<int>& arr, int l, int r) {
    if (l >= r) return;
    int mid = (l + r) / 2;
    mergeSort(arr, l, mid);
    mergeSort(arr, mid + 1, r);
    merge(arr, l, mid, r);
}

int main() {
    vector<int> arr = {5, 2, 9, 1, 5, 6, 3, 8};

    cout << "排序前:";
    for (int x : arr) cout << x << " ";
    cout << endl;

    mergeSort(arr, 0, arr.size() - 1);

    cout << "排序后:";
    for (int x : arr) cout << x << " ";
    cout << endl;

    return 0;
}

直接从后面替换插入的方法

java 复制代码
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
  int i=nums1.size()-1;
  n--;
  m--;
    while(n>=0){
        while(m>=0&&nums1[m]>nums2[n]){
            swap(nums1[i--],nums1[m--]);
        }
        swap(nums1[i--],nums2[n--]);
    }
    }
};
相关推荐
WolfGang00732123 分钟前
代码随想录算法训练营 Day50 | 图论 part08
数据结构·算法·图论
女生也可以敲代码1 小时前
AI时代下的50道前端开发面试题:从基础到大模型应用
前端·面试
aini_lovee2 小时前
多目标粒子群优化(MOPSO)双适应度函数MATLAB实现
人工智能·算法·matlab
yong99902 小时前
图像融合与拼接:完整MATLAB工具箱
算法·计算机视觉·matlab
春风不语5052 小时前
深入理解主成分分析(PCA)
算法
apollowing2 小时前
启发式算法WebApp实验室:从搜索策略到群体智能的能力进阶(二十二)
算法·启发式算法·web app
晚枫歌F3 小时前
最小堆定时器
数据结构·算法
Cosolar3 小时前
告别无脑循环:深入解析 ReWOO 与 Plan-and-Execute Agent 架构
人工智能·面试·全栈
Lumos_7773 小时前
Linux -- 线程
java·jvm·算法
七颗糖很甜4 小时前
“十五五”气象发展规划:聚焦五大核心任务
大数据·python·算法