给你两个按非递减顺序排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你合并 nums2 到 nums1 中,使合并后的数组同样按非递减顺序排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
错误代码
出错在 while 循环,if 判断顺序不同。
如果不先排除 p1 和 p2 等于 -1 的情况,那么 nums1[p1] 和 nums2[p2] 就会报错
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = m-1;
int p2 = n-1;
int x = m+n-1;
int max;
while(p1 >= 0 || p2 >= 0){
if(nums1[p1] > nums2[p2])
max = nums1[p1--];
else if(nums2[p2] > nums1[p1])
max = nums2[p2--];
else if(p1 == -1)
max = nums2[p2--];
else if(p2 == -1)
max = nums1[p1--];
nums1[x--] = max;
}
}
}
正确代码
逆向双指针,由于nums1数组后为空,每次遍历nums1和nums2数组中值最大的数,从后往前放入数组,因为两个数组本身是有序的,因此他们中最大的数,要么是nums1[p1] ,要么是nums[p2]。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = m-1;
int p2 = n-1;
int x = m+n-1;
int max;
while(p1 >= 0 || p2 >= 0){
if(p1 == -1)
max = nums2[p2--];
else if(p2 == -1)
max = nums1[p1--];
else if(nums1[p1] > nums2[p2])
max = nums1[p1--];
else
max = nums2[p2--];
nums1[x--] = max;
}
}
}
方法优化
nums2 数组不为空的时候,将 nums2 插入 nums1 中,由于两个数组都是顺序的,如果nums1 为空或者 nums1 小于 nums2 ,则直接往 nums1 后面插。
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = m-1;
int p2 = n-1;
int p = m+n-1;
while(p2 >= 0){
if(p1 >= 0 && nums1[p1] > nums2[p2])
nums1[p] = nums1[p1--];
else
nums1[p] = nums2[p2--];
p--;
}
}
}
不动脑子解法
直接把nums1 和nums2合并为一个数组,再调用数组自带函数 sort,进行排序
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for(int i=0; i<n; i++)
nums1[m+i] = nums2[i];
Arrys.sort(nums1);
}
}