版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
题目描述如下:
给定一个非负整数数组 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
思路如下:
按照题目描述,由于数组元素中奇数偶数各占一半,问题等价于:奇数下标的数组元素为奇数,偶数下标的数组元素为偶数。
于是遍历偶数下标的数组元素,若发现其值为奇数(不合要求),则开始遍历奇数下标的数组元素,寻找其中值为偶数的元素(也不合要求),将这两个元素的值交换,然后继续对偶数元素的遍历(下一次需要遍历奇数下标元素时,注意从上次进行过交换的奇数下标开始进行遍历,因为前面遍历过的部分一定无误,避免重复,这也是代码中index的作用)。时间复杂度O(n),空间复杂度O(n)。
代码如下:
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& A) {
int index = 1;
int tmp;
for (int i = 0; i < A.size(); i += 2) {
if (A[i] % 2 != 0) {
for (int j = index; j < A.size(); j += 2) {
if (A[j] % 2 == 0) {
tmp = A[i];
A[i] = A[j];
A[j] = tmp;
index = j+2;
break;
}
}
}
}
return A;
}
};