题目描述
输入一个整数数组,实现一个函数来调整该数组中数字的顺序,使得所有的奇数位于数组的前半部分,所有的偶数位于数组的后半部分,并保证奇数和奇数,偶数和偶数之间的相对位置不变。
题目解法
相对位置改变的写法
我看的《剑指Offer》第二版是没有要求相对位置不变的,而牛客网上是有这个要求的,后面的也会补充牛客网上的解法。
先写写剑指Offer上的思路:维护两个指针,第一个指针从前开始找偶数,第二个指针从后开始找奇数,然后交换奇偶数位置。
LeetCode 905和这题一样,只不过是偶数在前,奇数在后。
数组[1,2,3,4,5,6,7,8]
从头找到第一个偶数2和末尾第一个奇数7,交换位置为[1,7,3,4,5,6,2,8]
开头第二个偶数4和末尾第二个奇数5,交换位置为[1,7,3,5,4,6,2,8]
时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( 1 ) O(1) O(1)。
def reOrderArray(array):
if len(array) < 2: return array
left = 0
right = len(array) -1
while left < right:
# 注意这里也要加上 left < right
while left < right and array[left] & 1 != 0:
left += 1
while left < right and array[right] & 1 == 0:
right -= 1
if left < right:
array[left], array[right] = array[right], array[left]
return array
类插入排序的解法
若要求相对位置不变,我们可以参考插入排序的思路:
从头开始找到第二个奇数,然后插入到第一个奇数右边,其他元素相应地后移。
数组[1,2,3,4,5,6,7,8]
第一个奇数为1,置于数组首位;
找到第二个奇数3,插入到1右边,2后移,则有[1,3,2,4,5,6,7,8];
找到第三个奇数5,插入到3右边,[2,4]后移,则有[1,3,5,2,4,6,7,8];
找到第四个奇数7,插入到5右边,[2,4,6]后移,则有[1,3,5,7,2,4,6,8]。
时间复杂度 O ( n 2 ) O(n^2) O(n2),空间复杂度 O ( 1 ) O(1) O(1)。
def reOrderArray(array):
if len(array) < 2: return array
for i in range(len(array)):
if array[i] & 1 == 1:
value = array[i]
j = i - 1
while j >= 0 and array[j] & 1 == 0:
array[j+1] = array[j]
j -=1
array[j+1] = value
return array
以空间换时间
借助两个辅助数组,时间复杂度 O ( n ) O(n) O(n),空间复杂度 O ( n ) O(n) O(n)。
class Solution:
def reOrderArray(self, array):
# write code here
if len(array) < 2: return array
odd, even = [], []
for x in array:
if x & 1 == 1:
odd.append(x)
else:
even.append(x)
return odd + even
列表解析式
一行写完。
def reOrderArray(array):
return [x for x in array if x&1 == 1] + [x for x in array if x&1 ==0]
2019.7.18 补充LeetCode 905,修改解法一的循环条件