合并两个有序数组的思路
题目
https://leetcode.cn/problems/merge-sorted-array/description/
思路
双指针法(从后往前)
-
初始化两个指针p1和p2,分别指向nums1的前m个有效元素的末尾(即m - 1)和nums2的末尾(即n - 1)。同时,初始化一个指针p,指向合并后数组的末尾(即m + n - 1)。
-
比较nums1[p1]和nums2[p2]的值:
如果nums1[p1] >= nums2[p2],则将nums1[p1]放到nums1[p]的位置,然后p1减 1,p减 1。
如果nums1[p1] < nums2[p2],则将nums2[p2]放到nums1[p]的位置,然后p2减 1,p减 1。 -
重复上述比较和赋值操作,直到p2小于 0 或者p1小于 0 。
如果p2大于等于 0,说明nums2中还有剩余元素,将这些剩余元素依次复制到nums1的前面部分(从索引 0 开始)。因为nums1前面部分的m个元素在比较过程中可能已经被后移,所以此时nums1前面部分有足够空间容纳nums2的剩余元素。
而如果p1大于等于 0,nums1中剩余元素已经在正确位置,无需额外处理。
双指针法(从后往前)在实际操作中更为高效,因为它避免了在nums1前面部分频繁移动元素,时间复杂度为O(m + n),满足题目进阶要求。

Java 实现
public class MergeSortedArray {
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];
p1--;
} else {
nums1[p] = nums2[p2];
p2--;
}
p--;
}
// 如果nums2还有剩余元素,将其复制到nums1的前面部分
System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
}
public static void main(String[] args) {
MergeSortedArray solution = new MergeSortedArray();
int[] nums1 = {1, 2, 3, 0, 0, 0};
int m = 3;
int[] nums2 = {2, 5, 6};
int n = 3;
solution.merge(nums1, m, nums2, n);
for (int num : nums1) {
System.out.print(num + " ");
}
}
}
来自AI,未验证

浙公网安备 33010602011771号