题目
给定一个非负整数数组 A,返回一个数组,在该数组中, A 的所有偶数元素之后跟着所有奇数元素。
你可以返回满足此条件的任何数组作为答案。
示例:
输入:[3,1,2,4]
输出:[2,4,3,1]
输出 [4,2,3,1],[2,4,1,3] 和 [4,2,1,3] 也会被接受。
提示:
1 <= A.length <= 5000
0 <= A[i] <= 5000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-array-by-parity
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
小结:来一遍快速排序法中的第一轮迭代,让左边的数都是偶数,右边的数都是奇数。
class Solution(object):
def sortArrayByParity(self, A):
"""
:type A: List[int]
:rtype: List[int]
"""
tmp = A[0]
l = 0
r = len(A) - 1
while l < r:
while l < r and A[r] % 2 == 1:
r -= 1
A[l] = A[r]
while l < r and A[l] % 2 == 0:
l += 1
A[r] = A[l]
A[l] = tmp
return A