题目
给你两个按非递减顺序排列的整数数组nums1和nums2,另有两个整数 m 和 n ,分别表示nums1和nums2中的元素数目。
请你合并nums2到nums1中,使合并后的数组同样按非递减顺序排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为0 ,应忽略。nums2 的长度为n。
solution.1 合并之后再进行快排
快排的实现方式和原理
- 先手写一个快速排序的方法,使用了递归的方式,这里需要注意的是,pivot应该在if语句内部进行赋值,否则在nums1和nums2合并之后只有两个元素的情况下会有报错(超出索引,地址访问报错)。
- 然后我们在merge方法中用遍历的方式,将nums2插入到nums1的末尾,然后使用快速排序。
时间复杂度:
由于我们是对两个数组先进行合并然后再快速排序,因此,长度为m+n,所以时间复杂度为(m+n)log(m+n)。
空间复杂度:
由于快速排序是一种原地排序方式(并没有创建新的数组),因此空间复杂度套用快速排序的公式即log(m+n)。
class Solution {
public:void quickSort(vector<int>& vi, int lo, int hi){int pivot;int i = lo;int j = hi;if (lo < hi){pivot = vi[lo];while (i != j){while (vi[j] >= pivot && j > i){j--;}while (vi[i] <= pivot && j > i){i++;}if(i<j){swap(vi[i], vi[j]);}}swap(vi[lo], vi[i]);quickSort(vi, lo, i-1);quickSort(vi, i+1, hi);}}void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {for(int i = 0;i != n;i++){nums1[m+i] = nums2[i];}quickSort(nums1, 0, m+n-1);}
};
solution.2 双指针
- 使用两个分别指向nums1和nums2数组末端的指针(由于题目的设置是nums1后面补齐了0,为了给nums2的元素预留位置,因此这里的指向nums1数组末端的指针的真正含义是指向nums1最后一个不为0的值的位置),然后我们分别比较两个位置的元素的大小,从m+n-1的位置开始从后往前补齐元素。
时间复杂度:
即移动数据的次数,最多移动m+n次,即时间复杂度为m+n。
空间复杂度:
我们并没有创建新的数组而是在原地进行操作,因此空间复杂度为1。
class Solution {
public:void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {int p1 = m-1;int p2 = n-1;int p3 = m + n - 1;int cur = 0;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[p3--] = cur;}}
};