题目描述
给定两个排序后的数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-merge-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
从后往前遍历两个数组,将大的复制到A数组的后面,除了刚才比较小的那个数组的下标不动,另两个下标减一,向前遍历。
核心代码逻辑
void merge(int* A, int ASize, int m, int* B, int BSize, int n){
int i = m - 1;
int j = n - 1;
int k = m + n -1;
while(i >= 0 && j >= 0){
if(A[i] > B[j]){
A[k] = A[i];
i--;
} else {
A[k] = B[j];
j--;
}
k--;
}
while(j >= 0){
A[k] = B[j];
j--;
k--;
}
}
注意
- 这几个声明的下标如何移动
- 第一个while语句结束有两种情况
- 然后第二个while语句判断具体时间哪一种情况,然后做具体处理(A数组有剩余元素就不用管,因为他剩余的元素已经有序的排列在A中,如果是B数组有剩余元素,那么复制到A中即可)