题目:
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
思路:
利用双指针,分别指向两个数组,比较大小后放入一个新数组内暂存,最后把新数组的内容赋给nums1。
代码:
class solution{
public:void merge(vector<int>&nums1, int m, vector<int>nums2, int n){int p0=0,p1=0;int sorted[m+n];//防止nums1中的值被覆盖,建立暂存数组int cur;while(p0 < m || p1 < n){if(p0 == m){cur = nums2[p1++];}else if(p1 == n){cur = nums1[p0++];}else if(nums1[p0] < nums2[p1]){cur = nums1[p0++];}else{cur = nums2[p1++];}sorted[m+n-1] = cur;}for(int i = 0; i != m+n; i++){nums1[i] = sorted[i];}}
};
时间复杂度为:O(m+n)
空间复杂度为:O(m+n)
逆向双指针:
若是用双指针指向数组末尾,从后往前进行排序再从末尾填充进nums1,因为nums1的空间大小为m+n,恰好是够装nums1+nums2的内容的,所以也不会出现被覆盖的问题。
class solution{
public:void merge(vector<int>&nums1, int m, vector<int>&nums2, int n){int p1 = m-1, p2 = n-1;int cur;int tail = m + n - 1;while(p1 >= 0 || p2 >= 0){if(p1 == -1){cur = nums2[p2--];}else if(p2 == -1){cur = nums1[p1--];}else if(nums1[p1] > nums2[p2]){cur = nums1[p1--];}else{cur = nums2[p2--];}nums1[tail--] = cur;} }
};