题目
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.
翻译
假设有两个已经排序的数组,nums1和nums2。将nums2数组合并到nums1中,合并为一个排序数组。
注:
你可认为数组nums1具有足够的空间用来保存合并到其中的数组2的元素。
思路
如果采取归并排序中的方式,则需要开辟一个数组。然后从头遍历数组,然后归并。
这里因为第一个数组含有足够的空间,那么我们可以反其道行之。即从数组nums1和nums2尾部开始归并。然后将其中的较大值放到nums1尾部。直到归并结束。
代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int index1 = m-1;
int index2 = n -1;
int pos = m+n-1;
while (index2 >= 0) {
nums1[pos--] = index1 >= 0 && nums1[index1] > nums2[index2] ? nums1[index1--] : nums2[index2--];
}
}
};