题目:合并两个有序数组
给你两个有序整数数组 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]
错误实现:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int sorted[m+n];
int i=0,j=0,cur=0;
while(i<m||j<n){
if(nums1[i]<nums2[j]&&i<m){
sorted[cur]=nums1[i];
i++;
}
else{
sorted[cur]=nums2[j];
j++;
}
cur++;
}
for(int i=0;i<m+n;i++){
nums1[i]=sorted[i];
}
}
};
实现一:双指针 从前往后
//双指针 从前往后
//时间复杂度O(m+n) 空间复杂度O(m+n)需要开辟一个与nums1一样长度的数组的空间
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int sorted[m+n];
int i=0,j=0,cur=0;
while(i<m||j<n){
if(j==n){
//nums2全部加载完后,保证nums1的加载
//必须写在条件(nums1[i]<nums2[j])前,否则会出现数组溢出的情况
sorted[cur]=nums1[i++];
}
else if(i==m){
//nums2全部加载完后,保证nums1的加载
//必须写在条件(nums1[i]<nums2[j])前,否则会出现数组溢出的情况
sorted[cur]=nums2[j++];
}
else if(nums1[i]<nums2[j]){、
sorted[cur]=nums1[i++];
}
else{
sorted[cur]=nums2[j++];
}
cur++;
}
for(int i=0;i<m+n;i++){
nums1[i]=sorted[i];
}
}
};
实现二:逆向双指针 从后往前
//逆向双指针 从后往前
//不需要开辟额外的空间 空间复杂度为0(1)
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i=m-1,j=n-1;
for(int p=m+n-1;p>=0;p--){
if(j==-1){
nums1[p]=nums1[i--];
}
else if(i==-1){
nums1[p]=nums2[j--];
}
else if(nums1[i]>nums2[j]){
nums1[p]=nums1[i--];
}
else{
nums1[p]=nums2[j--];
}
}
}
};