注:
题目:
给你两个有序整数数组 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]
提示:
nums1.length = m + n
nums2.length = n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109
题解:
算法
nums1的后半部分是空的,可以直接覆盖而不会影响结果。因此可以指针设置为从后向前遍历,每次取两者之中的较大者放进 nums1的最后面。
严格来说,在此遍历过程中的任意一个时刻,nums1数组中有 m−p1−1 个元素被放入nums1的后半部,nums2数组中有 n−p2 −1 个元素被放入nums1的后半部,而在指针 p1 的后面,nums1数组有m+n−p1−1 个位置。
复杂度分析
时间复杂度:O(m+n)。指针移动单调递减,最多移动 m+n 次,因此时间复杂度为 O(m+n)。
空间复杂度:O(1)。直接对数组 nums1原地修改,不需要额外空间。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int index=m+n-1;
int p1=m-1;
int p2=n-1;
//当访问完任意一个数组时,跳出
while(p1!=-1&&p2!=-1){
if(nums1[p1]>nums2[p2]){
nums1[index]=nums1[p1];
p1--;
}
else{
nums1[index]=nums2[p2];
p2--;
}
index--;
}
//处理剩下的那个数组
if(p1==-1){
while(p2!=-1){
nums1[index]=nums2[p2];
p2--;
index--;
}
}
else{
while(p1!=-1){
nums1[index]=nums1[p1];
p1--;
index--;
}
}
return ;
}
};