LeetCode:数组之合并两个有序数组

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

解法:

1、使用C++自带的sort函数,先将nums2的值都复制到nums1的后面,然后直接使用sort函数进行排序即可

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        //将两个数组合并后再排序使用sort函数
        vector<int> vec;
        for(int i = 0;i < m;i++) vec.push_back(nums1[i]);
        nums1.clear();
        nums1 = vec;
        for(int i = 0; i < n;i++)nums1.push_back(nums2[i]);
        sort(nums1.begin(),nums1.end());
    }
};

2、双指针法

(1)从前到后

使用两个指针p1,p2分别指向数组nums1和nums2,然后另开辟一个数组vec,指针p指向vec,将nums1的值复制过去,然后比较nums1[p1]和nums2[p2]大小并赋值给vec[p]

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        //双指针法,从前往后
        int p1 = 0,p2 = 0,p = 0;
        vector<int> vec(nums1);
        while( (p1 < m)&&(p2 < n) )
        {
            nums1[p++] = vec[p1] <= nums2[p2] ? vec[p1++]:nums2[p2++]; 
        }
        if(p2 < n)
        {
            while(p2 < n)
            {
                nums1[p++] = nums2[p2++];
            }
        }
        if(p1 < m)
        {
            while(p1 < m)
            {
                nums1[p++] = vec[p1++];
            }
        }
    }
};

(2)从后到前

从nums1和nums2数组从后向前开始扫描,因为nums1数组后面没有值,所以这种方法可以节省空间。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        //双指针法,从后往前
        int p1 = m-1;//nums1的指针
        int p2 = n-1;//nums2的指针
        int p = m + n -1;

        while( (p1>=0)&&(p2>=0) )
        {
            nums1[p--] = nums1[p1] >= nums2[p2] ? nums1[p1--]:nums2[p2--];
        }
        if(p1 < 0)
        { //当nums1数组排完而nums2元素未被合并完
            while(p2 >= 0)
            {
                nums1[p--] = nums2[p2--];
            }
        }
    }
};
发布了80 篇原创文章 · 获赞 101 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/BigDream123/article/details/103330422