[leetcode]88. 合并两个有序数组

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

方法1:
使用优先队列priority_queue 将nums2中的元素构造成一个小顶堆minHeap,然后将每次用堆顶元素与nums1中的元素依次比较,如果堆顶元素小就将两个元素互换位置:堆顶元素加入到当前nums1中元素的位置,并将nums1中的元素压入堆中,重新构造小顶堆。如此一直到nums1中m的位置,从此之后只是将小顶堆的堆顶元素依次加入nums1即可。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (n == 0)
            return ;
        
        priority_queue<int, vector<int>, greater<int>> minHeap;
        for (int i = 0; i < n; i++)
            minHeap.push(nums2[i]);
        
        int ptr = 0;
        while (ptr < m) {
          if (minHeap.top() < nums1[ptr]) {
                int tmp = nums1[ptr];
                nums1[ptr] = minHeap.top();
                minHeap.pop();
                minHeap.push(tmp);
            }
            ptr++;
        }
        while (!minHeap.empty()) {
                nums1[ptr++] = minHeap.top();
                minHeap.pop();
        } 
    }
};

方法2:
双指针 / 从后往前

将指针p 置为 nums的(m+n-1)处,将指针p1 置为 nums1的(m)处, p2为 nums2的(n)处,在每一步将最大值放入nums1中p处。

注意:
如果当nums2倒叙遍历完毕(p2 = -1 < 0),此时nums1还未结束(p1 >= 0 && p >= 0),则剩余元素已经有序的存在nums1中,结束即可;
如果当nums1倒叙遍历完毕(p1 = -1 < 0),此时nums2还未结束(p2 = p >= 0),则将nums2中剩余元素(已经有序)复制到nums1中p之后的空间,结束即可;

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (n == 0)
            return ;
        if (m == 0) {
            nums1 = nums2;
            return ;
        }
            
        int p1 = m - 1;
        int p2 = n - 1;
        int p = m + n - 1;
        while (p1 >= 0 && p2 >= 0)
            nums1[p--] = nums1[p1] >= nums2[p2] ? nums1[p1--] : nums2[p2--];
        
        while (p2 >= 0){
            nums1[p2] = nums2[p2];
            p2--;
        }
            
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值