88. 合并两个有序数组 一遍遍历,减少临时变量

题目

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

说明:

初始化 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]

 解法:

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (0 == n)
            return;

        int index = m+n-1, mi = m-1, ni = n-1;
        for (; index >= 0 && mi >= 0 && ni >= 0; --index) {
            if (nums1[mi] > nums2[ni]) {
                nums1[index] = nums1[mi];
                --mi;
            } else {
                nums1[index] = nums2[ni];
                --ni;
            }  
        }
        while (ni >= 0)
            nums1[index--] = nums2[ni--];
    }
};

更精简的一种实现如下,不额外引入变量。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int len = nums1.size();
        if (len != m+n || n == 0)
            return;
        
        for (int i = len-1; n > 0 && m > 0; --i) {   //从结果数组的末尾开始
            if (nums2[n-1] > nums1[m-1]) {
                nums1[i] = nums2[n-1];
                --n;
            } else {
                nums1[i] = nums1[m-1];
                --m;
            }
        }
        while (n) {                                  //处理可能剩余的第二个数组
            nums1[n-1] = nums2[n-1];
            --n;
        }
    }
};

            复制代码会引起莫名其妙的错误,还不好排查,尽量自己手敲代码。注意临时变量的增减。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值