C++ - 合并两个有序数组为一个有序数组

分享一个大牛的人工智能教程。零基础！通俗易懂！风趣幽默！希望你也加入到人工智能的队伍中来！请点击http://www.captainbed.net

/*
 * MergeArray.cpp - by Chimomo
 *
 * 问题：有两个有序数组，合并成一个有序数组。
 * 分析：在内存中申请一个大小为两有序数组大小之和的新数组作为合并后的数组，从后往前把两个数组合并到新数组中。
 */

#include 
#include 
#include 

using namespace std;

#define MAX 1024

/**
 * 合并长度为length1的有序数组array1和长度为length2的有序数组array2为一个新有序数组并返回之。
 * @param arr1 The array 1.
 * @param arr2 The array 2.
 * @param len1 The length of array1.
 * @param len2 The length of array2.
 * @return The merged array.
 */
int *MergeArray(int *arr1, int *arr2, int len1, int len2) {
    int *mergedArray = new int[len1 + len2];
    if (arr1 == NULL || arr2 == NULL || (len1 + len2) > MAX) {
        return NULL;
    }

    int p = len1 + len2 - 1;
    int p1 = len1 - 1;
    int p2 = len2 - 1;
    while (p1 >= 0 && p2 >= 0) {
        if (arr1[p1] > arr2[p2]) {
            mergedArray[p--] = arr1[p1--];
        } else {
            mergedArray[p--] = arr2[p2--];
        }
    }
    while (p1 >= 0) {
        mergedArray[p--] = arr1[p1--];
    }
    while (p2 >= 0) {
        mergedArray[p--] = arr2[p2--];
    }

    return mergedArray;
}

int main() {
    int array1[6] = {1, 3, 7, 8, 9, 11};
    int array2[3] = {2, 5, 6};
    int *mergedArray = MergeArray(array1, array2, 6, 3);
    for (int i = 0; i <= 6 + 3 - 1; i++) {
        cout << *(mergedArray + i) << " ";
    }
    cout << endl;

    return 0;
}

// Output:
/*
1 2 3 5 6 7 8 9 11

*/