题目描述
给你两个有序整数数组 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 可以存储下两个数组中所有的元素,因此无需使用多余的数组,题目中给定的是两个有序数组,那么可以利用这个性质来遍历数组,起始指针的位置处于实际元素的最末尾位置,示例1 中 nums1 数组指针开始 i 指向3位置,nums2 指针 j 开始指向6位置,此外还需要一个指针来标记结果数组中遍历到的位置,起始位置为 m+n-1 处。
- 如果说指针 j 所指向的元素大于 i 指针指向的元素,就需要将大的元素放置到指针 k 处,j 向左移动,k 也向左移动,同理对于i 指针也是如此
- 需要注意的是如果两个数组遍历过程中其中一个数组提前遍历结束,那么需要将另一个数组的元素全部拷贝到剩下的位置中去。也就是 i 有可能等于-1,j 有可能等于-1
- 为什么不需要判断k>=0 呢?原因是两个数组遍历结束,k 恰好等于0,k-- 之后等于-1,但是循环已经结束了,所以不需要判断该条件。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int k = m+n-1;
if(!m) nums1=nums2;
int i=m-1,j=n-1;
while(i>=0 || j>=0)
{
if(i==-1) nums1[k--]=nums2[j--];
else if(j==-1) nums1[k--]=nums1[i--];
else if(nums2[j]>nums1[i]) nums1[k--]=nums2[j--];
else nums1[k--]=nums1[i--];
}
}
};
上述代码合并一下可以得到下面的代码。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int count=n+m-1;
n=n-1,m=m-1;
while(n>=0 || m>=0)
{
if(m>=0 && A[m]>B[n] || n<0) A[count--]=A[m--];
else A[count--]=B[n--];
}
}
};