合并排序数组

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

样例

给出A=[1,2,3,4]，B=[2,4,5,6]，返回 [1,2,2,3,4,4,5,6]

挑战

你能否优化你的算法，如果其中一个数组很大而另一个数组很小？

解题思路：这道题为了优化算法，我想的是把元素多的数组A向元素少的数组B中添加；遍历元素少的数组B，遇到A中比当前B中小的即可插进去，当B遍历完之后，将其余的A全部放在B的末尾即可。

class Solution {
    /**
     * @param A and B: sorted integer array A and B.
     * @return: A new sorted integer array
     */
    public ArrayList<Integer> mergeSortedArray(ArrayList<Integer> A, ArrayList<Integer> B) {
        // write your code here
        if(A.size()<B.size()){
            ArrayList<Integer>  tmp = A;
            A = B;
            B = tmp;
        };
        int i =0 ,j=0;
        while(i<A.size()&&j<B.size()){
            if(A.get(i)<B.get(j)){
                B.add(j,A.get(i));
                i++;
                j++;
            }else{
                j++;
            }
        }
        if(j==B.size()&&i<A.size()){
            while(i<A.size()){
                B.add(A.get(i));
                i++;
            }
        }
        return B;
    }
}