88. 合并两个有序数组
2018-12-02 本文已影响7人
闭门造折
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出:
[1,2,2,3,5,6]
思路
从后往前遍历两个数组,选择较大的放到nums1尾部空间中
性能分析
时间复杂度O(N),空间复杂度O(1)
具体代码
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 使用题目给的m和n作为尾索引
m--;
n--;
// 使用r(ight)表示新数组尾索引
int r = m + n + 1;
// 当两个数组均有剩余时
while(m >=0 && n >= 0){
// 选取两个尾部较大的一个,放入新数组
if(nums1[m] > nums2[n]){
nums1[r--] = nums1[m--];
}
else{
nums1[r--] = nums2[n--];
}
}
// 若nums1先全部插入完毕,则nums2剩余部分全部复制
if(m == -1){
while(n>= 0){
nums1[r--] = nums2[n--];
}
}
else{
while(m >= 0){
nums1[r--] = nums1[m--];
}
}
return;
}
};