922. Sort Array By Parity II（按奇偶排序数组 II）

922. 按奇偶排序数组 II

给定一个非负整数数组 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

 

解法一

//时间复杂度O(n), 空间复杂度O(1)
class Solution {
public:
    vector<int> sortArrayByParityII(vector<int>& A) {
        int n = A.size();
        int i = 0, j = 0, w = 0;
        while(i < n && A[i] % 2 != 0) i++;
        while(j < n && A[j] % 2 == 0) j++;
        bool even = true;
        while(w < n - 1) {
            if(even) {
                swap(A[w++], A[i++]);
                while(i < n && A[i] % 2 != 0) i++;
                while(j < n && A[j] % 2 == 0) j++;
            }
            else {
                swap(A[w++], A[j++]);
                while(i < n && A[i] % 2 != 0) i++;
                while(j < n && A[j] % 2 == 0) j++;
            }
            even = !even;
        }
        return A;
    }
};

解法二

//时间复杂度O(n), 空间复杂度O(n)
class Solution {
public:
    vector<int> sortArrayByParityII(vector<int>& A) {
        int n = A.size();
        vector<int> B(n);
        int i = 0, j = 1;
        for(int num : A) {
            if(num % 2 == 0) {
                B[i] = num;
                i += 2;
            }
            else {
                B[j] = num;
                j += 2;
            }
        }
        return B;
    }
};

解法一

使用三个指针w、i、j，w指向要写入的位置，i指向下一个偶数，j指向下一个奇数。偶-奇交替向w位置写入（需要用交换而不是直接覆盖），写入之后重新调整i、j指针的位置，直到w写完整个数组。

解法二

上一个解法i、j各遍历了一次数组，总共遍历了两次。解法二使用了一个额外的等长数组B来避免两次遍历。同样使用两个指针i、j，分别位向偶数位置，奇数位置。遍历一次数组A，遇到偶数就将其写入B中的i位置，遇到奇数就写到B中的j位置，直到遍历A完成，刚好写完B。

2019/08/06 11:27