LeetCode | Merge Sorted Array

来源:互联网 发布:怎样加入淘宝网店 编辑:程序博客网 时间:2024/06/05 03:44

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

Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.

一开始写的仔细想想似乎是O(N^2)的,因为需要插入n个元素,而每次需要对长为m进行遍历和移动数组元素的操作,确实是O(n^2).
看了discuss区之后发现,由于数组可以随机访问任意一个节点,可以考虑从数组末尾入手,倒序寻找最大的元素。
这样可以在O(m+n)的复杂度里找到所有的元素。

class Solution {public:    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {        //这是O(N^2)的解法        // //将n个数据插入到m个数据的nums1中        // for(int i=0;i<n;i++){        //     //找出插入的位置,利用二分查找        //     int left=0,right=m+i-1;        //     while(left<right){        //         int mid=(left+right)/2;        //         //找寻的值还比nums[i]小,则在左侧找        //         if(nums1[mid]<nums2[i]) left=mid+1;        //         else right=mid;        //     }        //     //这样可以找出right为第一个>nums[i]的位置,或者right为末尾        //     // if(nums1[right]>nums2[i])        //     // nums1.insert(right,nums2[i]);        //     // else nums1.push_back(nums2[i]);        //     // int pos=0;        //     // for(;pos<m+i && nums1[pos]<nums2[i];pos++);        //     //获得第一个比nums2[i]大的元素        //     for(int j=m+i;j>pos;j--)        //     nums1[j]=nums1[j-1];        //     //放入元素        //     nums1[pos]=nums2[i];        // }        //这是O(n)的        int i=m-1,j=n-1,k=m+n-1;        while(i>=0 && j>=0){            //从最后取值,如果nums1数据较大            if(nums1[i]>nums2[j])            nums1[k--]=nums1[i--];            else            nums1[k--]=nums2[j--];        }        while(j>=0)        nums1[k--]=nums2[j--];    }};;
0 0