LeetCode刷题_88:合并两个有序数组【C++】_leetcode 合并两个有效数组 c++-优快云博客

本文链接：https://blog.youkuaiyun.com/anchuanxu/article/details/98606844

博客介绍了如何使用C++解决LeetCode第88题，即合并两个有序数组的问题。提供了两种方法，一种是使用额外的空间，另一种是在不增加空间的情况下仅利用nums1完成合并。文章包含了详细的代码实现、测试用例以及性能对比。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

答案代码

方法一　用新增空间来做。

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

        int p = 0,q = 0;
        for(int i = 0; i < nums3.size(); i++){
            if(p < m && q < n){
                if(nums1[p] <= nums2[q])
                    nums3[i] = nums1[p++];
                else
                    nums3[i] = nums2[q++];
            }
            else if(p < m && q >= n){
                nums3[i] = nums1[p++];
            }
            else if(p >= m && q < n){
                nums3[i] = nums2[q++];
            }
        }
        for(int i = 0; i < nums1.size(); i++){
            nums1[i] = nums3[i];
        }

    }
};

方法二：不新增空间，只利用nums1

class Solution2 {
public:
    void merge2(vector<int>& nums1, int m, vector<int>& nums2, int n) {
    int p = m-1, q = n-1;
    for(int k = m+n-1; k >=0; k--)
    {
        if(p >= 0 && q >=0)
        {
            if(nums1[p] >= nums2[q])
                nums1[k] = nums1[p--];
            else
                nums1[k] = nums2[q--];
        }
        else if(p < 0 && q >= 0)
            nums1[k] = nums2[q--];
    }
    }

};

测试代码

#include <iostream>
#include <vector>

using namespace std;

//方法一　用新增空间来做。

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

        int p = 0,q = 0;
        for(int i = 0; i < nums3.size(); i++){
            if(p < m && q < n){
                if(nums1[p] <= nums2[q])
                    nums3[i] = nums1[p++];
                else
                    nums3[i] = nums2[q++];
            }
            else if(p < m && q >= n){
                nums3[i] = nums1[p++];
            }
            else if(p >= m && q < n){
                nums3[i] = nums2[q++];
            }
        }
        for(int i = 0; i < nums1.size(); i++){
            nums1[i] = nums3[i];
        }

    }
};

//方法二： 不新增空间，只利用nums1
class Solution2 {
public:
    void merge2(vector<int>& nums1, int m, vector<int>& nums2, int n) {
    int p = m-1, q = n-1;
    for(int k = m+n-1; k >=0; k--)
    {
        if(p >= 0 && q >=0)
        {
            if(nums1[p] >= nums2[q])
                nums1[k] = nums1[p--];
            else
                nums1[k] = nums2[q--];
        }
        else if(p < 0 && q >= 0)
            nums1[k] = nums2[q--];
    }
    }

};

int main() {
    int m ,n;
    cin>>m>>n;
    int arr1[m+n],arr2[n];
    for(int i = 0; i < m; i++)
        cin>>arr1[i];
    cout<<"初始化arr1结束"<<endl;
    cout<<"arr1中的元素为："<<endl;
    for(int i = 0; i < m; i++)
        cout<<arr1[i];
    cout<<endl;

    for(int i = 0; i < n; i++)
        cin>>arr2[i];
    cout<<"初始化arr2结束"<<endl;
    cout<<"arr2中的元素为："<<endl;
    for(int i = 0; i < n; i++)
        cout<<arr2[i];
    cout<<endl;

    vector<int> vec1(arr1,arr1+sizeof(arr1)/ sizeof(int));
    vector<int> vec2(arr2,arr2+sizeof(arr1)/ sizeof(int));

    //Solution().merge(vec1,m,vec2,n);
    Solution2().merge2(vec1,m,vec2,n);

    cout<<"合并的结果为： "<<endl;
    for(int i = 0; i < m+n; i++){
        cout<<vec1[i];
    }
    std::cout <<std::endl;
    return 0;
}