[每日一题]104:合并两个有序数组

170 阅读1分钟

文章目录


题目描述

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

提示:

-10^9 <= nums1[i], nums2[i] <= 10^9
nums1.length == m + n
nums2.length == n

题解思路

方法一:

  创建一个辅助数组,用来存储。

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

方法二:

  双指针,从后往前

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

方法三:

  sort排序

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        nums1.resize(m+n);
        for(int i = 0; i < n; i++) nums1[m+i]=nums2[i];
        sort(nums1.begin(), nums1.end());
    }
};

如有不同见解,欢迎留言讨论~~~