6. Merge Two Sorted Arrays

6. Merge Two Sorted Arrays

Description

Merge two given sorted integer array A and B into a new sorted integer array.

Example

A=[1,2,3,4]
B=[2,4,5,6]
return [1,2,2,3,4,4,5,6]

Solution

public class Solution {
    /**
     * @param A: sorted integer array A
     * @param B: sorted integer array B
     * @return: A new sorted integer array
     */
    public int[] mergeSortedArray(int[] A, int[] B) {
        // write your code here
        int len = A.length + B.length;
        int[] res = new int[len];
        int a=0,b=0,i=0;
        while(a<A.length&&b<B.length){
            if(A[a]<=B[b]) {
                res[i] = A[a];
                a++;
                i++;
            } else{
                res[i] = B[b];
                b++;
                i++;
            }
        }
        if(a>=A.length){
            while(b<B.length){
                res[i] = B[b];
                b++;
                i++;
            }
        }
        if(b>=B.length){
            while(a<A.length){
                res[i] = A[a];
                a++;
                i++;
            }
        }
        return res;
    }
}