1. 程式人生 > >leetcode 905. 按奇偶排序陣列【陣列】

leetcode 905. 按奇偶排序陣列【陣列】

題目:

給定一個非負整數陣列 A,返回一個由 A 的所有偶數元素組成的陣列,後面跟 A 的所有奇數元素。

你可以返回滿足此條件的任何陣列作為答案。

 

示例:

輸入:[3,1,2,4]
輸出:[2,4,3,1]
輸出 [4,2,3,1],[2,4,1,3] 和 [4,2,1,3] 也會被接受。

 

提示:

  1. 1 <= A.length <= 5000
  2. 0 <= A[i] <= 5000

思路:

  下面程式碼的時間複雜度和空間複雜度都是O(n).先申請一個長度和A一樣長度的list集合,設定分別指向list頭和尾的索引。然後遍歷一遍A,是偶數的話就從前面插入list,是基數的話就從後面插入list。

程式碼:

class Solution:
    def sortArrayByParity(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        res = []
        i = 0
        j = len(A)-1
        for item in A:
            if item % 2 == 0:
                res.insert(i,item)
                i += 1
            else:
                res.insert(j,item)
                j -= 1
        return res