给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。初始化 A 和 B 的元素数量分别为 m 和 n。
示例
输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
说明 A.length == n+m
链接: https://leetcode-cn.com/problems/sorted-merge-lcci/
解题思路
暴力算法
由于A后面没有存储数据的长度足以包含B。直接将B放进A,然后排序即可。
双指针算法
由于A数组和 B数组都是有序的,直接new一个新数组。比较A[fast] 和B[slow]的值(fast指向A,slow指向B),谁小就放到新空间,然后将指针移向下一位。(我这里是仿照合并两个有序链表做的,差不多一个意思)
public void merge(int[] A, int m, int[] B, int n) {
int[] arr = new int[A.length];
int fast = 0;
int slow = 0;
for (int i = 0; i < arr.length; i++) {
if (slow >= n && fast <= m) {
arr[i] = A[fast];
fast++;
continue;
}
if (slow <= n && fast >= m) {
arr[i] = B[slow];
slow++;
continue;
}
if (A[fast] <= B[slow]) {
arr[i] = A[fast];
fast++;
} else {
arr[i] = B[slow];
slow++;
}
}
System.arraycopy(arr, 0, A, 0, arr.length);
}
尾部双指针算法
由于A.length == n+m,可以直接将两个指针分别指向两个数组的尾部(这里的尾部指的是数组存放数据的尾部),然后放到A的尾部(这里指的是A的长度的尾部),就不用new新数组了。