[LeetCode] Merge Sorted Array

浣熊

于 2014-05-24 16:32:30 发布

阅读量705

点赞数

CC 4.0 BY-SA版权

分类专栏：面试算法

本文链接：https://blog.csdn.net/yuanbohx/article/details/26842489

算法同时被 2 个专栏收录

43 篇文章

订阅专栏

面试

40 篇文章

订阅专栏

本文介绍了一种高效的算法，用于将两个已排序的整数数组B合并到A中形成一个有序数组。该算法从后向前合并，避免了从前往后合并时需要不断后移元素的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

Given two sorted integer arrays A and B, merge B into A as one sorted array.

Note:
You may assume that A has enough space (size that is greater or equal to m + n) to hold additional elements from B. The number of elements initialized in A and B are m and nrespectively.

代码：

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int abIndex = m + n - 1;
        int aIndex = m - 1;
        int bIndex = n - 1;
        
        while(aIndex >= 0 && bIndex >= 0) {
            if(A[aIndex] > B[bIndex]) {
                A[abIndex--] = A[aIndex--];
            }
            else {
                A[abIndex--] = B[bIndex--];
            }
        }
        
        while(bIndex >= 0) {
            A[abIndex--] = B[bIndex--];
        }
    }
};

思路：

如果企图从头到尾把B中的元素插入到A中，插入位置后面的所有元素整体后移的办法，那效率确实有些低。

本文所示代码的算法是从后往前插入，开始插入的位置为m+n-1，由此保证插入过程不会覆盖掉A原有的元素值。