LeetCode刷题_88:合并两个有序数组【C++】

题目描述

给定两个有序整数数组 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& nums1, int m, vector& nums2, int n) {
        vector 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& nums1, int m, vector& 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 
#include 

using namespace std;

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

class Solution {
public:
    void merge(vector& nums1, int m, vector& nums2, int n) {
        vector 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& nums1, int m, vector& 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结束"<>arr2[i];
    cout<<"初始化arr2结束"< vec1(arr1,arr1+sizeof(arr1)/ sizeof(int));
    vector vec2(arr2,arr2+sizeof(arr1)/ sizeof(int));

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

    cout<<"合并的结果为： "<

性能对比

方法一

方法二

这就是强者的世界吗？

欢迎讨论，一起刷题，一起前往强者的世界。CSDN@anchuanxu 折剑山庄