LintCode刷题之路（六）：合并排序数组Ⅱ

最新推荐文章于 2022-03-16 16:50:11 发布

Molength

最新推荐文章于 2022-03-16 16:50:11 发布

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode

本文链接：https://blog.youkuaiyun.com/qq_38213612/article/details/80286531

lintcode 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种合并两个已排序整数数组A和B的方法，通过比较A和B中元素的大小，将它们按顺序放入新的数组中。文章还讨论了当其中一个数组远大于另一个时的优化策略。

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

合并两个排序的整数数组A和B变成一个新的数组。
样例
给出A=[1,2,3,4]，B=[2,4,5,6]，返回 [1,2,2,3,4,4,5,6]
挑战
你能否优化你的算法，如果其中一个数组很大而另一个数组很小？

C++

class Solution {
public:
    /**
     * @param A: sorted integer array A
     * @param B: sorted integer array B
     * @return: A new sorted integer array
     */
    vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
        // write your code here
        vector<int> result(A.size()+B.size());
        int i = 0 ,j = 0;
        int k = 0;

        while(i<A.size()&&j<B.size())
        {
            if(A[i]<B[j])
            {
                result[k] = A[i];
                k++;
                i++;
            }
            else if(A[i]>B[j])
            {
                result[k] = B[j];
                k++;
                j++;
            }
            else
            {
                result[k] = A[i];
                k++;
                i++;
                result[k] = B[j];
                k++;
                j++;
            }
        }

        while(i<A.size())
        {
            result[k] = A[i];
            k++;
            i++;
        }

        while(j<B.size())
        {
            result[k] = B[j];
            k++;
            j++;
        }

        return result;
    }
};

Py3：

class Solution {
public:
    /**
     * @param A: sorted integer array A
     * @param B: sorted integer array B
     * @return: A new sorted integer array
     */
    vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
        // write your code here
        vector<int> result(A.size()+B.size());
        int i = 0 ,j = 0;
        int k = 0;

        while(i<A.size()&&j<B.size())
        {
            if(A[i]<B[j])
            {
                result[k] = A[i];
                k++;
                i++;
            }
            else if(A[i]>B[j])
            {
                result[k] = B[j];
                k++;
                j++;
            }
            else
            {
                result[k] = A[i];
                k++;
                i++;
                result[k] = B[j];
                k++;
                j++;
            }
        }

        while(i<A.size())
        {
            result[k] = A[i];
            k++;
            i++;
        }

        while(j<B.size())
        {
            result[k] = B[j];
            k++;
            j++;
        }

        return result;
    }
};