LeetCode刷题之路 按奇偶排序数组 II

按奇偶排序数组 II【简单】

给定一个非负整数数组 A, A 中一半整数是奇数,一半整数是偶数。

对数组进行排序,以便当 A[i] 为奇数时,i 也是奇数;当 A[i] 为偶数时, i 也是偶数。

你可以返回任何满足上述条件的数组作为答案。

示例:

输入:[4,2,5,7]
输出:[4,5,2,7]
解释:[4,7,2,5],[2,5,4,7],[2,7,4,5] 也会被接受。

提示:

  1. 2 <= A.length <= 20000
  2. A.length % 2 == 0
  3. 0 <= A[i] <= 1000

这道题比较简单,直接将奇数偶数全部挑出来,然后进行一个循环按顺序加进去就好了。注意:简单的列表创建不要用循环,尽量用列表推导,因为列表推导不仅更具有你可读性,而且效率相对更高。

class Solution:
    def sortArrayByParityII(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        odd = [i for i in A if i % 2]
        even = [i for i in A if not i % 2]
        s = []
        while(odd and even):
            s.append(even.pop())
            s.append(odd.pop())
        return s

推荐阅读更多精彩内容