题目
给定两个有序整数数组 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]
题解
- 两个数组都是有序的,设定三个指针,p1指向nums1有数字的最后一位,p2指向nums2最后一位,p指向nums1最后一位;
- 然后从后往前进行遍历比较两个数组中数字的大小,大的方到nums1[p],然后继续往前比较,注意指针值的变化就好;
- 另外最后当p1<0时,还有nums2中的数字没匹配到,说明这些数是最小的,将其copy到nums1头部,长度为p2+1,详细请看代码。
代码
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 (p1 >= 0 && p2 >= 0) {
if (nums1[p1] > nums2[p2]) {
nums1[p] = nums1[p1];
p--;
p1--;
} else {
nums1[p] = nums2[p2];
p--;
p2--;
}
}
System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
}
}