给定一个非负整数数组
nums
,nums
中一半整数是 奇数 ,一半整数是 偶数 。对数组进行排序,以便当
nums[i]
为奇数时,i
也是 奇数 ;当nums[i]
为偶数时,i
也是 偶数 。你可以返回 任何满足上述条件的数组作为答案 。
这道题是 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;
- }
- };