题目
给定一个非负整数数组 nums
, nums
中一半整数是 奇数 ,一半整数是 偶数 。
对数组进行排序,以便当 nums[i]
为奇数时,i
也是 奇数 ;当 nums[i]
为偶数时, i
也是 偶数 。
你可以返回 任何满足上述条件的数组作为答案 。
示例 1:
输入:nums = [4,2,5,7]
输出:[4,5,2,7]
解释:[4,7,2,5],[2,5,4,7],[2,7,4,5] 也会被接受。
示例 2:
输入:nums = [2,3]
输出:[2,3]
提示:
2 <= nums.length <= 2 * 10^4
nums.length
是偶数nums
中一半是偶数0 <= nums[i] <= 1000
解题方法
构造数组
构造额外数组,遍历原数组,分别将奇数和偶数按顺序填在奇数位和偶数位。
时间复杂度O(n)
,空间复杂度O(n)
。
C++代码:
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& nums) {
vector<int> result(nums.size(), 0);
int oddidx = 0;
int evenidx = 1;
for(int num : nums) {
if(num%2==0) {
result[oddidx] = num;
oddidx += 2;
}
else {
result[evenidx] = num;
evenidx += 2;
}
}
return result;
}
};
双指针
双指针分别遍历奇数位和偶数位,当在偶数位遇到奇数时与奇数位遇到的第一个偶数交换。
时间复杂度O(n)
,空间复杂度O(1)
C++代码:
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& nums) {
int idx = 1;
for(int i=0; i<nums.size(); i+=2) {
if(nums[i]%2) {
while(nums[idx]%2 != 0) idx += 2;
swap(nums[idx], nums[i]);
}
}
return nums;
}
};