lintcode：合併排序數組

題目：

合併排序數組

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

樣例

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

挑戰

你可否優化你的算法，若是其中一個數組很大而另外一個數組很小？

解題：

利用Java的ArrayList很簡單的，時間複雜度O(n+m)兩個數組都要遍歷一遍，對應兩個數組長度差異很大的狀況，效率就低了

Java程序：

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
        ArrayList merge = new ArrayList();
        int aSize = A.size();
        int bSize = B.size();
        int i=0;
        int j = 0;
        while(i<aSize && j<bSize){
            if(A.get(i)<=B.get(j)){
                merge.add(A.get(i));
                i++;
            }else{
                merge.add(B.get(j));
                j++;
            }
        }
        while(i<aSize){
            merge.add(A.get(i));
            i++;
        }
        while(j<bSize){
            merge.add(B.get(j));
            j++;
        }
        return merge;
    }
}

View Code

總耗時: 1832 ms

這個好無節操的哦

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
        int bSize = B.size();
        for(int i=0;i<bSize;i++)
            A.add(B.get(i));
        Collections.sort(A);
        return A;
    }
}

View Code

總耗時: 1340 ms

Python程序：

Python也能夠無節操的來

class Solution:
    #@param A and B: sorted integer array A and B.
    #@return: A new sorted integer array
    def mergeSortedArray(self, A, B):
        # write your code here
        A = A + B 
        A.sort()
        return A 

View Code

總耗時: 454 ms

固然也能夠複雜的來了

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

View Code

總耗時: 314 ms