letecode [88] - 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]
题目大意:
给定排序数组nums1和nums2,合并两数组于数组nums1中,按照排序存放元素。
理 解 :
合并后元素的大小为 k = m+n,为排序时不移动前面的元素,从后向前排序。
比较nums1[m-1]和nums2[n-1],将大的值放在后面,对应指针前移。若nums1[m-1] > nums2[n-1],即nums1[k-1] = nums1[m-1],--k ; --m。
当nums1全被比较但nums2还有剩余时,将剩余nums2元素直接放在nums1前面。
代 码 C++:
class Solution { public: void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) { if(n==0) return; if(m==0){ for(int i=0;i<n;++i) nums1[i] = nums2[i]; return; } int k = m+n-1; while(m>0 && n>0){ if(nums1[m-1]>=nums2[n-1]){ nums1[k] = nums1[m-1]; --m; --k; }else{ nums1[k] = nums2[n-1]; --n; --k; } } while(n>0){ nums1[k] = nums2[n-1]; --n; --k; } } };
运行结果:
执行用时 : 4 ms 内存消耗 : 8.8 MB