【LintCode 簡單】6. 合併排序數組 II

1.問題描述:

合併兩個排序的整數數組A和B變成一個新的數組。


2.樣例:

給出A=[1,2,3,4]B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]。


3.代碼:

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
        if A is None:
            return B
        if B is None:
            return A
        l=[]
        a=0
        b=0
        alength=len(A)
        blength=len(B)
        while a<alength and b<blength:
            if A[a]<=B[b]:
                l.append(A[a])
                a+=1
            else:
                l.append(B[b])
                b+=1
        if a!=alength:
            for i in range(a,alength):
                l.append(A[i])
        if b!=blength:
            for i in range(b,blength):
                l.append(B[i])
        return l
            
        



發佈了79 篇原創文章 · 獲贊 1 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章