题目:
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
解释:需要合并 [1,2,3] 和 [2,5,6] 。
合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
思路:
创建一个长度为m+n的新数组nums用于存放结果 int三个指针 i、j、k分别指向nums1、nums2和nums; 如果i指针指到了末尾,那么就把指针j 后面指着的数全部放入nums;j也是一样;
比较nums[i]和nums[j],把小的数放入nums中,指针后移
代码:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
vector<int> nums(m + n);
int i=0,j=0,k=0;
int cur = 0;
while(k < m + n)
{
if(i == m)
{
cur = nums2[j++];
}
else if(j == n)
{
cur = nums1[i++];
}
else if(nums1[i] < nums2[j])
{
cur = nums1[i++];
}
else
{
cur = nums2[j++];
}
nums[k] = cur;
k++;
}
nums1 = nums;
}
};