双指针法
题目详情
给你两个按 非递减顺序 排列的整数数组 nums1
和 nums2
,另有两个整数 m
和 n
,分别表示 nums1
和 nums2
中的元素数目。
请你 合并 nums2
到 nums1
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1
的初始长度为 m + n
,其中前 m
个元素表示应合并的元素,后n
个元素为 0 ,应忽略。nums2
的长度为n
。
示例1:
输入: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 中的元素。
示例2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
解释:需要合并 [1] 和 [] 。
合并结果是 [1] 。
示例3:
输入:nums1 = [0], m = 0, nums2 = [1], n = 1
输出:[1]
解释:需要合并的数组是 [] 和 [1] 。
合并结果是 [1] 。
注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
我的代码:
C++
class Solution
{
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
int p = m-- + n-- -1;
while(m>=0&&n>=0)
nums1[p--]=nums1[m]>nums2[n]?nums1[m--]:nums2[n--];
while(n>=0)
nums1[p--]=nums2[n--];
}
};
Java
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//p用来索引nums1即将被插入的位置(从后往前)
//nums1和nums2合并后的数组长度为m+n,末尾元素的索引为m+n-1
int p = m + n - 1;
//注意m,n两个指针遍历的时候是从索引为m-1和n-1开始的
--m;
--n;
while (m >= 0 && n >= 0){
//谁大先插入排后面
nums1[p--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
}
//将nums2剩余元素放入nums1
while (n >= 0){
nums1[p--] = nums2[n--];
}
}
}
涉及知识点:
1.双指针算法
双指针主要用于遍历数组,两个指针指向不同的元素,从而协同完成任务。也可以延伸到多个数组的多个指针。
若两个指针指向同一数组,遍历方向相同且不会相交,则也称为滑动窗口(两个指针包围的区域即为当前的窗口),经常用于区间搜索。
若两个指针指向同一数组,但是遍历方向相反,则可以用来进行搜索,待搜索的数组往往是排好序的。
思路:
利用双指针法(实际还需要第三个),利用两个指针索引两个数组的末尾并往前遍历,因为是要合并到nums1的,所以nums2有剩余时候还要利用第三个指针来将剩余元素复制到nums1,若nums1有剩余,不用管,因为要的就是它.
ps:代码中多次用到了自增自减来使代码更简洁