88. 合并两个有序数组
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
说明:
初始化 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 {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int index = m + n - 1;
m--;
n--;
while (m >= 0 && n >= 0) {
if (nums1[m] > nums2[n]) {
nums1[index--] = nums1[m--];
} else {
nums1[index--] = nums2[n--];
}
}
//如果一个数组已经遍历完了,将另一个数组中的元素直接替换nums1之前的元素即可
if (m >= 0) {
while (m >= 0) {
nums1[index--] = nums1[m--];
}
} else {
while (n >= 0) {
nums1[index--] = nums2[n--];
}
}
}
}