合并排序的数组
题目:
给定两个排序后的数组 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两数组均为已经排好序的数组,要求将B数组合并入A数组,并且A有足够的空间去存放B。
- 想法1:
既然A有足够的空间,干脆将B直接添加在A的后面,再用冒泡排序或者快速排序等方法对A数组排序,想法实现起来很容易,可总有些浪费了题目设定中的两个排序后的数组 A 和 B,这里不给出代码。 - 想法二:
利用数组AB 已经排好序的特点,设立三个指针,index ,flag1, flag2 ,index在A数组中从最大长度ASize处后往前移动,flag1在A数组中从A中所存数字的最大位置处往前移动,flag2在B数组中从B中所存数字的最大位置处往前移动.
初始位置如图:
所以,如下所示通过比较flag1与flag2的大小,将值大的赋值给index,再将index与此时的较大的flag分别减1.
状态如下:
此时最大值6已经归位,继续循环,直到flag1或者flag2小于零,此时只需将flag不为零的数组剩余元素复制到A数组后面即可。
整个过程如下图:
(cur为index idx1为flag1 idx2为flag2)
代码如下:
void merge(int* A, int ASize, int m, int* B, int BSize, int n){
int index = m + n - 1;
int flag1 = m - 1;
int flag2 = n - 1;
while(index >= 0 && flag1 >= 0 && flag2 >= 0){
if(A[flag1] >= B[flag2]){
A[index--] = A[flag1--];
}else{
A[index--] = B[flag2--];
}
}
while(index >= 0 && flag1 >= 0){
A[index--] = A[flag1--];
}
while(index >= 0 && flag2 >= 0){
A[index--] = B[flag2--];
}
}