#yyds干货盘点# LeetCode程序员面试金典:合并排序的数组

时间:2023-02-13 19:16:39

题目:

给定两个排序后的数组 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]

代码实现:

class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int pa = 0, pb = 0;
int[] sorted = new int[m + n];
int cur;
while (pa < m || pb < n) {
if (pa == m) {
cur = B[pb++];
} else if (pb == n) {
cur = A[pa++];
} else if (A[pa] < B[pb]) {
cur = A[pa++];
} else {
cur = B[pb++];
}
sorted[pa + pb - 1] = cur;
}
for (int i = 0; i != m + n; ++i) {
A[i] = sorted[i];
}
}
}