Python Leetcode(905.按奇偶排序数组)
给定一个非负整数数组 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
Solution:(最开始的思路是遍历数组,对元素进行判断,奇数的话添加在结果列表后面,偶数的话插入在结果列表前面。)
class Solution(object):
def sortArrayByParity(self, A):
"""
:type A: List[int]
:rtype: List[int]
"""
result = []
for each in A:
if each % 2 == 0:
result.insert(0, each)
else:
result.append(each)
return result
solution = Solution()
print(solution.sortArrayByParity([1, 3, 5, 4, 6, 7]))
[6, 4, 1, 3, 5, 7]
Solution:(看了题解之后,发现这道题可以理解成按奇偶性排序,利用双指针,前指针遇到奇数停下,后指针遇到偶数停下,然后交换。)
class Solution(object):
def sortArrayByParity(self, A):
"""
:type A: List[int]
:rtype: List[int]
"""
start, end = 0, len(A)-1
while start < end:
if A[start] % 2 == 0:
start += 1
elif A[end] % 2 == 1:
end -= 1
else:
A[start], A[end] = A[end], A[start]
return A
solution = Solution()
print(solution.sortArrayByParity([1, 3, 5, 4, 6, 7]))
[6, 4, 5, 3, 1, 7]