1. 程式人生 > >Leetcode922.Sort Array By Parity II按奇偶排序陣列2

Leetcode922.Sort Array By Parity II按奇偶排序陣列2

給定一個非負整數陣列 A, A 中一半整數是奇數,一半整數是偶數。

對陣列進行排序,以便當 A[i] 為奇數時,i 也是奇數;當 A[i] 為偶數時, i 也是偶數。

你可以返回任何滿足上述條件的陣列作為答案。

示例:

輸入:[4,2,5,7] 輸出:[4,5,2,7] 解釋:[4,7,2,5],[2,5,4,7],[2,7,4,5] 也會被接受。

提示:

  1. 2 <= A.length <= 20000
  2. A.length % 2 == 0
  3. 0 <= A[i] <= 1000
class Solution {
public:
    vector<int> sortArrayByParityII(vector<int>& A)
    {
        int len = A.size();
        int cnt1 = len - 1;
        int cnt2 = len - 1;
        for(int i = 0; i < len; i++)
        {
            if((A[i] & 1) == 0 && (i & 1) == 1)
            {
                for(int j = cnt1; j > i; j--)
                {
                    if((A[j] & 1) == 1 && (j & 1) == 0)
                    {
                        swap(A[i], A[j]);
                        cnt1 = j - 1;
                        break;
                    }
                }
            }
            else if((A[i] & 1) == 1 && (i & 1) == 0)
            {
                for(int j = cnt2; j > i; j--)
                {
                    if((A[j] & 1) == 0 && (j & 1) == 1)
                    {
                        swap(A[i], A[j]);
                        cnt2 = j - 1;
                        break;
                    }
                }
            }
        }
        return A;
    }
};