6. 合并排序数组 II【简单】

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

思路:双指针移动比较,最后把未处理的数都加进去vector即可。

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

 

posted @ 2018-01-26 16:21  陈阿毛  阅读(82)  评论(0)    收藏  举报