[LeetCode] 88_合并两个有序数组
题目要求
给定两个有序整数数组 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]
题目分析
- 以num1为基础合并,从后往前,分三个指针,k指向合并后的数组,i指向原nums1数组,j指向原nums2数组。每次从两个数组中选取较大值放到最后面,k和被选取的数组指针向前进1。循环条件是两个数组都没有到头。
- 当有一个原数组到头时,结束循环。
- 这时如果nums2数组还没有循环到头,那么将nums2中剩余的数全部依次填充。
注意点
- 从后往前而不是从前往后,这样可以不会覆盖到还没选取的nums1中的数。
- 如果是nums1没有循环完的话不需要进行填充,因为nums1中的数本来就是nums1中。
代码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int k=m+n-1,i=m-1,j=n-1;
while(i>=0 &&j>=0){
if(nums1[i]>=nums2[j]){
nums1[k--]=nums1[i--];
}
else{
nums1[k--]=nums2[j--];
}
}
while(j>=0){
nums1[k--]=nums2[j--];
}
}
};