【LeetCode & 剑指offer刷题】查找与排序题5:Merge Sorted Array

【LeetCode & 剑指offer 刷题笔记】目录(持续更新中...)

Merge Sorted Array

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]

C++
 
//问题:合并两个有序数组
//方法一:开辟临时数组,通过比较两个有序数组,复制到临时数组中,再把排好序的数组复制回原数组
/*class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
    {
       
        int i=0,j=0,k=0;
        vector<int> temp(m+n);
        while(i<m&&j<n)
        {
            temp[k++] = (nums1[i]<nums2[j])? nums1[i++]:nums2[j++];
        }
        while(i<m)
        {
            temp[k++] = nums1[i++];
        }
        while(j<n)
        {
            temp[k++] = nums2[j++];
        }
       
        int size = nums1.size();
        if(size >= (m+n))
            nums1 = temp;//将排序好元素赋值过去
        else
        {
            nums1.assign(temp.begin(), temp.begin()+size); //仅仅赋值部分元素
            cout<<"nums1空间不足!!"<<endl;
        }
       
    }
};*/
//方法二:从后往前扫描,依次复制(未开辟额外空间,效率更优)
class Solution
{
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
    {
        int i=m-1,j=n-1,k=nums1.size()-1;//从后往前扫描, i,j分别指向有序数组末尾,k指向nums1空间的末尾
        while(i>=0 && j>=0)
        {
            nums1[k--] = (nums1[i] > nums2[j])? nums1[i--]:nums2[j--];
        }
        while(j >= 0) //将nums2剩余元素复制过去
        {
            nums1[k--] = nums2[j--];
        }
       
    }
};
 
 

 

posted @ 2019-01-05 20:09  wikiwen  阅读(134)  评论(0编辑  收藏  举报