LeetCode: 88. 合并两个有序数组(C++)

题目:

给定两个有序整数数组 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]

解答:

这个题可以考虑一下新开一个m+n那么大的数组,然后不断地将两个数组中间较小的那个放前面,循环下去。

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
	    vector<int> result;
        int i = 0, j = 0;
        while (i < m || j < n) {
            while(i < m && j < n) {
                if (nums1[i] < nums2[j]) {
                    result.push_back(nums1[i]);
                    i++;
                }
                else if (nums1[i] == nums2[j]) {
                    result.push_back(nums1[i]);
                    result.push_back(nums2[j]);
                    i++;
                    j++;
                }
                else {
                    result.push_back(nums2[j]);
                    j++;
                }
            }
            if (i >= m && j < n) {
                result.push_back(nums2[j]);
                j++;
            }
            if (i < m && j >= n) {
                result.push_back(nums1[i]);
                i++;
            }
        }
        nums1.assign(result.begin(), result.end());
	}
};

然而,还有更好的方法:

由于数组num1的空间已经足够,其实是可以不用再浪费空间的。、

那么,可以从数组的最尾部向前一次插入,将比较大的元素放进num1的末尾,最后如果num2有剩余在把num2省的属继续插入,num1有剩余不用管,已经完成了。

下面是代码:

class Solution {
public:
	void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int i = m - 1, j = n - 1, k = m + n - 1;
        while (i >= 0 && j >= 0) {
            if (nums1[i] >= nums2[j]) {
                nums1[k] = nums1[i];
                k--;
                i--;                
            }
            else {
                nums1[k] = nums2[j];
                k--;
                j--;
            }
        }
        while (j >= 0) {
            nums1[k] = nums2[j];
            k--;
            j--;
        }
                 
	}
};

ps:更简洁的代码(来自LeetCode):

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
         int i = m - 1, j = n - 1, index = m + n - 1;
                   while (i >= 0 && j >= 0)
                   {
                             nums1[index--] = nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
                   }
                   while (j >= 0)
                             nums1[index--] = nums2[j--];

    }
};

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值