输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有奇数在数组的前半部分,所有偶数在数组的后半部分。
示例:
输入:nums = [1,2,3,4]
输出:[1,3,2,4]
注:[3,1,2,4] 也是正确的答案之一。
提示:
0 <= nums.length <= 50000
0 <= nums[i] <= 10000
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/diao-zheng-shu-zu-shun-xu-shi-qi-shu-wei-yu-ou-shu-qian-mian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
AC代码
class Solution:
def exchange(self, nums: List[int]) -> List[int]:
prior = 0
tail = len(nums) - 1
while prior < tail:
if nums[prior] & 1:
prior += 1
if not (nums[tail] & 1):
tail -= 1
if prior >= tail:
break
if (not (nums[prior] & 1)) and (nums[tail] & 1):
nums[prior], nums[tail] = nums[tail], nums[prior]
prior += 1
tail -= 1
return nums
笔者直接上的双指针,不过需要注意一个问题,也是WA仔细想了代码之后才发现的:在while循环中,笔者处理完prior和tail之后其实是需要再进行一次循环条件判定的,不然在前后两半交界处容易把相邻的奇偶调换导致解答错误。
有启发的解法
class Solution {
public int[] exchange(int[] nums) {
int left = 0;
int right = nums.length - 1;
while (left < right) {
while (left < right && nums[left] % 2 != 0) {
left++;
}
while (left < right && nums[right] % 2 == 0) {
right--;
}
if (left < right) {
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
}
return nums;
}
}
笔者写的时候就觉得应该一次性把前指针推进到偶数位置,但脑子秀逗了就想着if,while循环嵌套就行,这样子还可以省略上面那个憨憨的判定,并且只需要考虑大小即可,时间复杂度会低于笔者最开始写的方法。