数组-合并排序数组 II-简单

题目描述

合并两个排序的整数数组A和B变成一个新的数组。

样例
给出A=[1,2,3,4]，B=[2,4,5,6]，返回 [1,2,2,3,4,4,5,6]
挑战
你能否优化你的算法，如果其中一个数组很大而另一个数组很小？

题目链接

分析

对于合并两个已经排好序的数组问题，直接使用归并排序的思想。当两个数组比较的时候，选择较小的那个数，并将其压入返回的数组里面，然后移动到下一个位置。当一个数组被比较完时，直接将另一个数组全部压入返回数组即可。最终的时候复杂度O(n)，空间复杂度O(n)。

程序

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> C;
        int size_A = A.size();
        int size_B = B.size();
        int i = 0, j = 0;
        while(i < size_A && j < size_B){
            if(A[i]<B[j]){
                C.push_back(A[i++]);
            }
            else{
                C.push_back(B[j++]);
            }
        }
        while(i < size_A){
            C.push_back(A[i++]);
        }
        while(j < size_B){
            C.push_back(B[j++]);
        }
        return C;
    }
};