题目
给定两个排序后的数组 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和B共有元素n个,那么目的就是有序的填慢A中的前n个空。
- 用end表示A中要填满区间的尾index。endA表示A中元素的尾index, endB表示B的尾index。
- 同时从尾部遍历A和B。end > -1为循环终止条件。
- 当endA等于-1时,说明A中的元素遍历完了。若B中还有元素,填充到A中。
- 当endB等于-1时,说明B中的已经填完了,此时直接break。
- 当endA和endB都不为-1时,选择大的从尾部填充。
代码
class Solution {
public:
void merge(vector<int>& A, int m, vector<int>& B, int n) {
int end = m + n - 1;
int endA = m - 1;
int endB = n - 1;
while ( end > -1 ) {
if ( endA == -1 )
A[end--] = B[endB--];
else if ( endB == -1 )
break;
else {
if ( A[endA] >= B[endB] )
A[end--] = A[endA--];
else
A[end--] = B[endB--];
}
}
return;
}
};
来源:CSDN
作者:PandaDou
链接:https://blog.csdn.net/m0_37822685/article/details/104627689