题目描述
给你一个整数数组 nums,将 nums 中的的所有偶数元素移动到数组的前面,后跟所有奇数元素。
返回满足此条件的 任一数组 作为答案。
输入与输出
输入:nums = [3,1,2,4]
输出:[2,4,3,1]
解释:[4,2,3,1]、[2,4,1,3] 和 [4,2,1,3] 也会被视作正确答案。
提示
- 1 <= nums.length <= 5000
- 0 <= nums[i] <= 5000
思路
利用双指针实现,类似于快速排序思想,从左边扫直到遇到奇数为止,从右边扫直到遇到偶数为止,然后交换这两个数。
AC代码(C++)
class Solution {
public:
vector<int> sortArrayByParity(vector<int>& nums) {
int n = nums.size();
int l = 0, r = n - 1;
while(l < r) {
while(l < r && nums[l] % 2 == 0) ++l;
while(l < r && nums[r] % 2 == 1) --r;
if (l < r) {
swap(nums[l], nums[r]);
}
}
return nums;
}
};
AC代码(Java)
class Solution {
public int[] sortArrayByParity(int[] nums) {
int n = nums.length;
int l = 0, r = n - 1;
while(l < r) {
while(l < r && nums[l] % 2 == 0) ++l;
while(l < r && nums[r] % 2 == 1) --r;
if (l < r) {
nums[l] = nums[l] ^ nums[r];
nums[r] = nums[l] ^ nums[r];
nums[l] = nums[l] ^ nums[r];
}
}
return nums;
}
}
AC代码(Python)
class Solution:
def sortArrayByParity(self, nums: List[int]) -> List[int]:
n = len(nums)
l, r = 0, n - 1
while l < r:
while l < r and nums[l] % 2 == 0:
l+=1
while l < r and nums[r] % 2 == 1:
r-=1
if l < r:
nums[l], nums[r] = nums[r], nums[l]
return nums