题目:
给定两个有序整数数组 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]
分析:
方法一:刚开始想创建一个新的数组,长度等于m+n。之后比较两个数组的每个元素,将小的存入新的数组,这种方法当其中一个数组遍历完成时,另一个数组会有剩余的元素,需要将其手动添加到后面的元素。
package LanQiao;
import java.util.Arrays;
public class Main {
public static void main(String[] args) {
int[] nums1 = { 1, 2, 3, 0, 0, 0 };
int[] nums2 = { 2, 5, 6 };
merge1(nums1, 3, nums2, 3);
}
public static void merge(int[] nums1, int m, int[] nums2, int n) {
int[] num = new int[m + n];
int i = 0, j = 0, k = 0;
while (i < m && j < n) {
if (nums1[i] <= nums2[j])
num[k++] = nums1[i++];
else
num[k++] = nums2[j++];
}
System.out.println(i);
System.out.println(j);
while (i < m)
num[k++] = nums1[i++];
while (j < n)
num[k++] = nums2[j++];
for (int k1 = 0; i < num.length; k1++) {
System.out.println(num[k1]);
}
}
}
这种方法原理上没有错,但是在LeetCode上却运行错误。是因为题目上要求将 nums2 合并到 nums1 中,而我是将其赋值给新的数组。
方法二:将nums2 合并到 nums1 后面,之后使用排序方法即可。
package LanQiao;
import java.util.Arrays;
public class Main {
public static void main(String[] args) {
int[] nums1 = { 1, 2, 3, 0, 0, 0 };
int[] nums2 = { 2, 5, 6 };
merge1(nums1, 3, nums2, 3);
}
public static void merge1(int[] nums1, int m, int[] nums2, int n) {
if (m <= 0) {
nums1[0] = nums2[0];
}
for (int i = 0; i < nums2.length; i++) {
nums1[m + i] = nums2[i];
}
Arrays.sort(nums1);
for (int i = 0; i < nums1.length; i++) {
System.out.println(nums1[i]);
}
}
}
对数组排序可以使用自己写的冒泡排序,也可以使用Arrays类中封装好的sort方法,并且这种方法在速度上也有较大的提升。
运行结果: