给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
方法一:直接合并后并排序
算法:
最直观的方法是先将数组nums2放进数组nums1的尾部,然后直接对整个数组进行排序。
代码:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for(int j = m,i=0;j<m+n;j++,i++){
nums1[j]=nums2[i];
}
Arrays.sort(nums1);
}
}
//时间复杂度 套用快速排序的时间复杂度O((m+n)log(m+n))
//空间复杂度 套用快排 O(log(m+n))
方法二:双指针
算法:采用类似于归并排序的思想。
两个有序的数组,通过双指针的方法,进行比较,将较小的数放置在另外开辟的数组中,比较结束后,将新开辟的数组值复制到nums1中
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//采用双指针p1和p2
int p1=0,p2=0;
int[] temp = new int[m+n];
int cur;
while(p1<m || p2 <n ){
if(p1 == m){
cur = nums2[p2];
p2++;
}else if(p2 == n){
cur = nums1[p1];
p1++;
}else if(nums1[p1] < nums2[p2]){
cur=nums1[p1];
p1++;
}else{
cur = nums2[p2];
p2++;
}
temp[p1+p2-1] = cur;//进行赋值
}
for(int j = 0; j<m+n;j++){
nums1[j] = temp[j];
}
}
}
//时间复杂度:O(m+n)
//空间复杂度:O(m+n)
方法三:逆向双指针
算法:上述方法需要开辟一个新的数组,浪费空间。由于nums1数组后半部分是空的,所以我们可以逆向双指针来进行赋值。从后面的数进行比较,取较大的数放入nums1的后面。
这里我们需要考虑一下,是否存在nums1的前半部分被覆盖的情况。
不妨设nums1的逆向指针为p1(初始为m-1,p1>=0),nums2的逆向指针为p2(初始为n-1,p2>=0);假设nums1前部分的较小的数被nums2中较大的数覆盖,此时nums1数组p1指针后面填充了m+n-1-p1个数,其中nums1填入了m-1-p1个数,nums2填入了n-1-p2个数,若覆盖的话,则nums1和nums2填入的个数应该大于p1指针后面的数。也即:
解得: ,矛盾! 故不可能会覆盖掉。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//采用双指针p1和p2
//从后面开始遍历 节省空间
int p1=m-1,p2=n-1;
int p = m+n-1;//填充的指针
int cur;
while(p1>=0 || p2 >=0 ){
if(p1 == -1){
cur = nums2[p2];
p2--;
}else if(p2 == -1){
cur = nums1[p1];
p1--;
}else if(nums1[p1] > nums2[p2]){//取较大的数放置在最后面
cur = nums1[p1];
p1--;
}else{
cur = nums2[p2];
p2--;
}
nums1[p]=cur;
p--;
}
}
//时间复杂度O(m+n)
//空间复杂度O(1) 直接对nums1数组操作不需要额外空间
}