python - LintCode 6. 合并排序数组 II

6. 合并排序数组 II  

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

样例

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

class Solution:
    """
    @param A: sorted integer array A
    @param B: sorted integer array B
    @return: A new sorted integer array
    """
    def mergeSortedArray(self, A, B):
        # write your code here
        result = []
        i = 0
        j = 0
        while i<len(A) or j<len(B):
            if i == len(A):
                result.append(B[j])
                j += 1
            elif j == len(B):
                result.append(A[i])
                i += 1
            elif A[i] < B[j]:
                result.append(A[i])
                i +=1
            else:
                result.append(B[j])
                j+=1
        return result