给定一个非负整数数组 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
这个题我考虑多了,居然用了O(n)的空间来解决它。实际上利用双指针就可以解决了。
class Solution { public: vector<int> sortArrayByParity(vector<int>& A) { int size = A.size(); vector<int> ret(size,0); int start = 0, end = size -1 ; for (int i = 0; i <size; i++){ if(A[i] % 2 == 0){ ret[start++] = A[i]; }else{ ret[end--] = A[i]; } } return ret; } };
还有一种类似于partition的解法,原地改变数组并返回。
class Solution { public: vector<int> sortArrayByParity(vector<int>& A) { int left=0,right=A.size()-1; while(left<right){ while(left<right &&A[left]%2==0) left++; while(left<right &&A[right]%2) right--; if(left<right) swap(A[left++],A[right--]); } return A; } };