1. 题目
给定一个非负整数数组
nums
,nums
中一半整数是 奇数 ,一半整数是 偶数 。对数组进行排序,以便当
nums[i]
为奇数时,i
也是 奇数 ;当nums[i]
为偶数时,i
也是 偶数 。你可以返回 任何满足上述条件的数组作为答案 。
2. 示例
3. 分析
这道题是 905. 按奇偶排序数组 的第二个版本。对于这道题目,该文章有做详细解答:905. 按奇偶排序数组 - 力扣。题目中有使用额外空间和原地操作两种思路。对于本题来说,也是这两种思路。
新开辟一个数组res,大小为nums数组的大小。第一次遍历数组将偶数替换到res的偶数下标下,即res[0],res[2],res[4] ... ;第二次遍历数组将奇数替换到res的奇数下标下,即res[1],res[3],res[5] ...
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& nums) {
int n = nums.size();
vector<int> res(n);
for(int i = 0, j = 0; i < n; i++)
{
if(nums[i] % 2 == 0)
{
res[j] = nums[i];
j += 2;
}
}
for(int i = 0, j = 1; i < n; i++)
{
if(nums[i] % 2 == 1)
{
res[j] = nums[i];
j += 2;
}
}
return res;
}
};
其实也可以做到一次遍历即可。定义两个指针 j, k 。遍历数组元素若为偶数,则通过 j 指针将元素替换到res数组偶数下标下;为奇数则通过 k 指针将元素替换到res数组奇数下标下:
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& nums) {
int n = nums.size();
vector<int> res(n);
int j = 0, k = 1;
for(int i = 0; i < n; i++)
{
if(nums[i] % 2 == 0)
{
res[j] = nums[i];
j += 2;
}
else
{
res[k] = nums[i];
k += 2;
}
}
return res;
}
};
原地替换:i 指向偶数下标(0, 2, 4, ...),j 指向奇数下标(1, 3, 5, ...)。遍历数组若i指向的元素为奇数,则 j 需不断移动找到偶数元素,此时 nums[ i ] 就为奇数,nums[ j ] 就为偶数,交换二者即可。
class Solution {
public:
vector<int> sortArrayByParityII(vector<int>& nums) {
int n = nums.size();
int j = 1;
for(int i = 0; i < n; i += 2)
{
if(nums[i] % 2 == 1)
{
while(nums[j] % 2 == 1) j += 2;
swap(nums[i], nums[j]); // 此时的j指向的元素为偶数
}
}
return nums;
}
};