力扣面试题 10.01. 合并排序的数组 C语言实现

合并排序的数组

题目：

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