题目
输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有奇数位于数组的前半部分,所有偶数位于数组的后半部分。
示例:
输入:nums = [1,2,3,4]
输出:[1,3,2,4]
注:[3,1,2,4] 也是正确的答案之一。
提示:
- 1 <= nums.length <= 50000
- 1 <= nums[i] <= 10000
思路
看到这个题目的时候,我首先想到的就是 快速排序 方式。因为奇数和偶数两种特性很容易进行区分。
使用两个指针,分别指向头和末尾。当满足某项条件的时候,指针进行移动。
快速排序
快速排序的解法,其实也是双指针思想的一种应用。
这个解法参考了之前写的 LeetCode 75. 颜色分类 的三路快排的解法。
class Solution {
public int[] exchange(int[] nums) {
if(nums == null || nums.length == 0) return new int[0];
int left = 0;
int right = nums.length - 1;
int curr = 0;
// 可以使用快速排序的思想,如果是奇数,就排在左边。偶数右边
while(curr < right){
// 奇数
if((nums[curr]&1) == 1){
swap(nums, left, curr);
curr++;
left++;
}
// 偶数
if((nums[curr]&1) == 0){
// 交换两个数字
swap(nums, curr, right);
right--;
}
}
return nums;
}
public void swap(int[] nums, int index1, int index2){
int temp = nums[index1];
nums[index1] = nums[index2];
nums[index2] = temp;
}
}
这个解法是有瑕疵的,对于 75. 颜色分类那道题目(对三个元素进行划分),这种方法非常适合。
但是这道题,只对 两种 元素进行区分,所以有更简单的解法
更简单的解法
也是采用双指针的思想,分别指向数组的两端,也是快速排序的思想。不过这样的解法更契合 这篇文章 中的快速排序方法。
- 如果左边已经满足条件,就 left++
- 如果右边已经满足条件,就 right–
- 当前面两者的条件都不满足的时候,交换 left和 right下标处的值
代码
class Solution {
public int[] exchange(int[] nums) {
int left = 0;
int right = nums.length - 1;
while(left < right){
// 左边已经是奇数, left++
while(left < right && (nums[left]&1) == 1) left++;
// 右边已经是偶数, right--
while(left < right && (nums[right]&1) == 0) right--;
// 退出上面两个循环的时候,有两种情况
// 1 左边是偶数,右边是奇数
// 2 left == right
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
return nums;
}
}
明确一点,while(left < right) 导致最终退出循环的时候,left 必定等于 right