题目:合并两个有序数组
给定两个有序整数数组 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]
复制代码
思考:
因为两个数组本身有序,并且合并后nums1长度等于m+n。
所以可以从两个数组最后一个元素开始比较,将较大的元素添加到nums1最后,依次循环比较从后向前添加到nums1即可。
如果其中一个数组先添加完,继续将另一个数组剩下的元素,从后向前添加到nums1即可。
复制代码
实现:
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
//合并后数组最后一个元素下标
int index = m + n - 1;
//nums1最后一个元素下标
int i = m - 1;
//nums2最后一个元素下标
int j = n - 1;
//若第一个数组没有元素,直接将第二个数组中元素挨个加入
if (i < 0) {
while (j >= 0) {
nums1[index--] = nums2[j--];
}
}
//从后往前比较nums1和nums2的元素,将大的添加到nums1末尾
while (i >= 0 && j >= 0) {
if (nums1[i] > nums2[j]) {
nums1[index--] = nums1[i--];
} else {
nums1[index--] = nums2[j--];
}
}
//当一个数组添加完后,讲另一个数组剩下元素按个添加
if (i >= 0) {
while (i >= 0) {
nums1[index--] = nums1[i--];
}
}
if (j >= 0) {
while (j >= 0) {
nums1[index--] = nums2[j--];
}
}
}
}复制代码