题目
给定一个非负整数数组 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] 也会被接受。
思路
设置k1初始化为第一个偶数位置,k2初始化为第一个奇数位置。遍历整个数组,如果元素在正确的位置上,则可以忽略,继续向后移动;否则,如果元素是奇数,则将其放在k2指向的奇数位置上,并将k2移动到下一个奇数位置,如果元素是偶数,则将其放在k1指向的偶数位置上,并将k1移动到下一个偶数位置。在遍历过程中,一旦有k1(k2)的位置超过数组长度,就说明所有偶数(奇数)元素已经都放置在了正确的位置上,又因为题目给出偶数和奇数个数各占一半,所以奇数(偶数)元素也肯定都放在正确的位置上,此时可以结束遍历。
vector<int> sortArrayByParityII(vector<int>& A) {
//k1表指向偶数位置,k2指向奇数位置
int i = 0, k1 = 0, k2 = 1;
//注意循环条件
while(i < A.size() && k2 < A.size() && k1 < A.size())
{
//如果元素在正确的位置上,忽略,继续向后
if(i%2 == A[i]%2)
i++;
//不在正确的位置上
else{
//如果元素是奇数
if(A[i]%2)
{
swap(A[i],A[k2]);
k2 += 2;
}
//如果元素是偶数
else
{
swap(A[i],A[k1]);
k1 += 2;
}
}
}
return A;
}