题目
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解题
这个其实很简单, 就是对两个数组进行一次归并即可, 为了避免元素被覆盖, 从后向前进行归并就可以解决问题, 代码如下:
class Solution {
// 从后往前进行merge, 避免覆盖元素
public void merge(int[] nums1, int m, int[] nums2, int n) {
int index = m + n -1;
while(m > 0 || n > 0){
// n == 0表示数组2已经merge完成, num1不需要再进行merge了
if(n == 0){
return;
} else if(m == 0){
nums1[index] = nums2[n - 1];
n--;
} else if(nums1[m-1] > nums2[n-1]){
nums1[index] = nums1[m - 1];
m--;
} else {
nums1[index] = nums2[n - 1];
n--;
}
index--;
}
}
}