1. 程式人生 > >【LeetCode】905. 按奇偶校驗排序陣列

【LeetCode】905. 按奇偶校驗排序陣列

給定一個非負整數陣列 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 {
public:
    vector<int> sortArrayByParity(vector<int>& A) {
        int l = A.size();
        vector<int> ans(l);
        int a = 0;
        int b = l-1;
        for(int i = 0; i < l; i++)
        {
            if (A[i] % 2 == 0)
            {
                ans[a] = A[i];
                a++;
            }
            else
            {
                ans[b] = A[i];
                b--;
            }
        }
        return(ans);
    }
};