题目描述
Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
The number of elements initialized in nums1 and nums2 are m and n respectively.
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.
Example:
Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
Output: [1,2,2,3,5,6]
思路
双指针p1、p2,分别指向nums1和nums2最后一个元素(不算空元素)
还有一个指针 p 指向nums1最后一个位置
先逐个比较p1和p2上的元素,直到其中一个到达数组头:
若p2上的元素大于p1上的元素:把p2上的元素放到p的位置上,再将p2和p指针前移
若p2上的元素小于p1上的元素:把p1上的元素放到p的位置上,再将p1和p指针前移
注意:可能出现nums1元素都放进nums1了,但是nums2的元素还没放完的情形,
比如 nums1: [0] 0 , nums2:[1] 1
nums2没放完说明剩下的都是较小的元素,直接放进nums1最前面即可
实现
public void merge(int[] nums1, int m, int[] nums2, int n) {
//双指针分别指向nums1和nums2最后一个元素(不算空元素)
//从后面开始比较,比较大的从nums1最后开始放进nums1里
int p1=m-1; //指向nums1最后一个非空元素
int p2=n-1; //指向nums2最后一个元素
int p=m+n-1; //指向nums1最后一个位置
//逐个比较p1和p2上的元素,直到其中一个到达数组头
while(p1>=0 && p2>=0){
if(nums2[p2]>nums1[p1]){ //若p2比p1大
nums1[p]=nums2[p2]; //把p2上的元素放到p的位置上
p2--; //再将p2和p指针前移
p--;
}
else{ //若p2比p1小
nums1[p]=nums1[p1]; //把p1上的元素放到p的位置上
p1--; //再将p1和p指针前移
p--;
}
}
//如果nums1元素都放进nums1了,但是nums2的元素还没放完,说明这些剩下的都是最小的,直接逐个放进nums1最前面
while(p>=0 && p2>=0){
nums1[p]=nums2[p2];
p2--;
p--;
}
}