Lintcode: 整數排序 II

問題:

給一組整數,請將其在原地按照升序排序。使用歸併排序,快速排序,堆排序或者任何其他 O(n log n) 的排序算法。

樣例:

例1:

輸入:[3,2,1,4,5],
輸出:[1,2,3,4,5]。

例2:

輸入:[2,3,1],
輸出:[1,2,3]。

python:(不能通過lintcode的測試!!!)

class Solution:
    """
    @param A: an integer array
    @return: nothing
    """
    def mergeInt(self, left, right):
        result = []
        while left and right:
            if left[0] <= right[0]:
                result.append(left[0])
                del left[0]
            else:
                result.append(right[0])
                del right[0]
        if left:
            result += left
        if right:
            result += right
        return result
    def sortIntegers2(self, A):
        # write your code here
        if len(A) <= 1:
            return A
        mid = len(A)//2
        left = A[:mid]
        right = A[mid:]
        left = self.sortIntegers2(left)
        right = self.sortIntegers2(right)
        return self.mergeInt(left, right)

C++:

class Solution {
public:
    /**
     * @param A: an integer array
     * @return: nothing
     */
    void merge(vector<int> &result, vector<int> &left, vector<int> &right)
    {
        int lenLeft = left.size();
        int lenRight = right.size();
        int pL = 0;
        int pR = 0;
        while(pL < lenLeft && pR < lenRight)
        {
            if(left[pL] < right[pR])
            {
                result.push_back(left[pL++]);
            }else{
                result.push_back(right[pR++]);
            }
        }
        while(pL < lenLeft)
        {
            result.push_back(left[pL++]);
        }
        while(pR < lenRight)
        {
            result.push_back(right[pR++]);
        }
    }
    void sortIntegers2(vector<int> &A) {
        // write your code here
        int lenA = A.size();
        if(lenA <= 1)
        {
            return;
        }
        vector<int> left;
        vector<int> right;
        for(int i = 0; i < lenA; i++)
        {
            if(i < lenA/2)
            {
                left.push_back(A[i]);
            }else{
                right.push_back(A[i]);
            }
        }
        sortIntegers2(left);
        sortIntegers2(right);
        A.clear();
        merge(A, left, right);
    }
};

PS:用的是歸併排序

PS:python版本一直測試不過,但是在其他IDE已經沒有問題了。我自己覺得代碼沒問題,誰如果能發現問題請告知!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章