leetcode 922. 按奇偶排序数组 II(Sort Array By Parity 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] 也会被接受。
提示:
2 <= A.length <= 20000
A.length % 2 == 0
0 <= A[i] <= 1000
解法:
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int sz = A.size();
int odd = sz-1, even = sz-2;
int i = 0;
while(i < sz){
if(i%2 != A[i]%2){
if(i%2 == 1){
// A[i] is even
swap(A[i], A[even]);
even -= 2;
}else{
swap(A[i], A[odd]);
odd -= 2;
}
}else{
i++;
}
}
return A;
}
};

更多精彩