LeetCodeDay17 —— 合并两个有序数组

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]

思路

1. 利用一个临时数组保存有序结果，然后在拷贝回nums1
2. 将空间复杂度降低为O(1)，与在数组中间插入一个元素的思想类似，从后往前来。将最大的元素放在num1[m+n-1]的位置。

Tips

• 想到初步解后不要立即查询最优解，要多思考，至少要有一个是否能否更优解的判断。
• 一般来说要从3个角度去思考，时间复杂度、空间复杂度、代码简洁程度。

class Solution_88_01 {
 public:
  void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
    vector<int> nums;
    int i = 0, j = 0;
    while (i < m && j < n) {
      if (nums1[i] < nums2[j]) {
        nums.push_back(nums1[i]);
        ++i;
      } else {
        nums.push_back(nums2[j]);
        ++j;
      }
    }
    while (i < m) nums.push_back(nums1[i++]);
    while (j < n) nums.push_back(nums2[j++]);
    for (int i = 0; i < nums.size(); ++i) nums1[i] = nums[i];
  }
};

class Solution_88_02 {
 public:
  void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
    int len = m + n - 1;
    --m;
    --n;
    while (m >= 0 && n >= 0)
      nums1[len--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
    // while (m >= 0) nums1[len--] = nums1[m--]; 可以不需要
    while (n >= 0) nums1[len--] = nums2[n--];
  }
};