LeetCode 面试题 ☞ 合并两个有序数组

合并两个有序的数组

题目描述

• 题目

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

解题

标签:从后向前数组遍历

• 因为 nums1 的空间都集中在后面，所以从后向前处理排序的数据会更好，节省空间，一边遍历一边将值填充进去
• 指针 m 和 n 分别指向 nums1 和 nums2 的有数字尾部，从尾部值开始比较遍历，同时设置指针size 指向 nums1 的最末尾，每次遍历比较值大小之后，则进行填充
• 当 m<0 时遍历结束，此时 nums2 中还有数据未拷贝完全，将其直接拷贝到 nums1 的前面，最后得到结果数组
• 时间复杂度：O(m+n)O(m+n)

代码实现

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int size=m+n-1;

        while(n){
            if(m==0){
                nums1[size--]=nums2[--n];
            }else if(nums1[m-1]<nums2[n-1]){
                nums1[size--]=nums2[--n];
            }else{
                nums1[size--]=nums1[--m];
            }
        }
    }
};

原题🔗