[leetcode] 88. Merge Sorted Array 合并有序数组

最新推荐文章于 2025-07-18 15:47:59 发布

原创最新推荐文章于 2025-07-18 15:47:59 发布 · 176 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一种将两个已排序的整数数组合并成一个有序数组的算法。通过两种方法实现：一种是简单粗暴的合并后排序；另一种是更高效的合并排序写法，使用额外的数组进行元素比较和插入，最后将结果赋回原数组。

题目描述：

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.
Example:
Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

Output: [1,2,2,3,5,6]

暴力解法：

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

合并排序的写法，注意vector的拷贝用swap或者迭代的方法，两个vector进行比较插入一个新的vector：

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        nums1.erase(nums1.begin() + m, nums1.end());
        vector<int> nums3;
        int i, j;
        for(i = 0, j = 0; i < nums1.size() && j < nums2.size(); )
        {
            if(nums1[i] < nums2[j])
            {
                nums3.push_back(nums1[i]);
                i++;
            }
            else
            {
                nums3.push_back(nums2[j]);
                j++;
            }
        }
        if(i < nums1.size())
        {
            for(; i < nums1.size(); i++)
                nums3.push_back(nums1[i]);
        }
        if(j < nums2.size())
        {
            for(; j < nums2.size(); j++)
                nums3.push_back(nums2[j]);
        }
        swap(nums1, nums3);
    }
};