题目来源:https://leetcode-cn.com/problems/diao-zheng-shu-zu-shun-xu-shi-qi-shu-wei-yu-ou-shu-qian-mian-lcof/
输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有奇数位于数组的前半部分,所有偶数位于数组的后半部分。
示例:
输入:nums = [1,2,3,4]
输出:[1,3,2,4]
注:[3,1,2,4] 也是正确的答案之一。
其实是个简单题。。但是想记录一下不同的思路。
解法一:遍历+额外空间
第一反应就是遍历,建立一个长度相同的数组res,遍历整个nums,找到奇数就加到res的左边,找到偶数就加到res的右边。很直观的做法:
class Solution(object):
def exchange(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
n = len(nums)
res = [0] * n
odd = 0
even = n-1
for i in range(n):
if nums[i] % 2 == 1:
res[odd] = nums[i]
odd += 1
elif nums[i] % 2 == 0:
res[even] = nums[i]
even -= 1
return res
时间复杂度:O(n),一次循环
空间复杂度:O(n),开辟了一个大小为n的数组res,还有两个指针,所以是O(n+2)即O(n)
解法二:双指针
定义两个指针low和high,low从nums左边开始,找到第一个偶数,high从nums右边开始,找到第一个奇数,交换这两个位置的数,继续循环,直到low=high
class Solution(object):
def exchange(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
n = len(nums)
low = 0
high = n-1
while low < high:
while nums[low] % 2 == 1 and low < high:
low += 1
while nums[high] % 2 == 0 and low <high:
high -= 1
nums[low], nums[high] = nums[high], nums[low]
return nums
参考: