题目描述:
给定一个非负整数数组 A,返回一个由 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
解题过程:
我刚开始是这么写,输出居然有错误,让我很费解:
class Solution:
def sortArrayByParity(self, A: List[int]) -> List[int]:
for i in A:
if i%2==1:
A.remove(i)
A.append(i)
return A
换了个写法,正确了:
class Solution:
def sortArrayByParity(self, A: List[int]) -> List[int]:
for i in A:
if i%2==0:
A.remove(i)
A.insert(0,i)
return A
这种问题肯定能用列表表达式的,于是看评论区写法:
class Solution:
def sortArrayByParity(self, A: List[int]) -> List[int]:
return sorted(A,key=lambda x:(x%2!=0))
总结:
- insert可以用于往列表中制定位置插入元素;
- sorted可以传入参数key用于指定元素进行排列,我推测非指定元素统统前置。