题目
给定一个非负整数数组 A, A 中一半整数是奇数,一半整数是偶数。
对数组进行排序,以便当 A[i] 为奇数时,i 也是奇数;当 A[i] 为偶数时, i 也是偶数。
你可以返回任何满足上述条件的数组作为答案。
示例:
输入:[4,2,5,7]
输出:[4,5,2,7]
解释:[4,7,2,5],[2,5,4,7],[2,7,4,5] 也会被接受。
提示:
2 <= A.length <= 20000
A.length % 2 == 0
0 <= A[i] <= 1000
解答
1. 暴力解法
class Solution:
def sortArrayByParityII(self, A):
L1=[]#奇数
L2=[]#偶数
for i in A:
if i%2==0:
L2.append(i)
else:
L1.append(i)
L1.sort()
L2.sort()#不排序也可
for i in range(len(A)):
if i%2==0:
A[i]=L2.pop()
else:
A[i]=L1.pop()
return A
将数组中的奇数偶数分离到两个数组,再将两个数组的内容向原数组的奇数偶数位填充。
执行结果
执行用时:264ms
内存消耗:16.1MB
2. 暴力解法的优化
class Solution:
def sortArrayByParityII(self, A):
L1=[]#奇数
L2=[]#偶数
for i in A:
if i%2==0:
L2.append(i)
else:
L1.append(i)
ans=[]
for i in range(len(L1)):
ans.append(L2[i])
ans.append(L1[i])
return ans
第二次for
循环只需要循环len(A)/2
次,每次向空列表ans
中先添加偶数,再添加一个奇数
3. 双指针解法
class Solution:
def sortArrayByParityII(self, A):
n = len(A)
i, j = 0, 1 # i - even pointer, j - odd pointer
while i < n and j < n:
if A[i] % 2 == 0:
i += 2
continue
if A[j] % 2 == 1:
j += 2
continue
A[i], A[j] = A[j], A[i]
return A
两个序号(也可理解为指针),一个从0开始,每次加2,代表偶数下标位置,另一个从1开始,每次加2,代表奇数下标位置。
如果奇数下标的数也是奇数,则走到下一个奇数下标位置,同理,若偶数下标上也是偶数,则走向下一个偶数下标,如果两个指针所对应的下标的奇偶和数字的奇偶同时不同(奇数位置为偶数,偶数位置为奇数同时发生),则交换这两个数,循环这个步骤,直到有一个指针大于数组长度即可。
由于奇数偶数各占一半,若奇数位置到达最后一位,那么说明所有奇数已经都在奇数位置上,那么此时的偶数也都在偶数位置上。
4. 官方解法
class Solution(object):
def sortArrayByParityII(self, A):
N = len(A)
ans = [None] * N
t = 0
for x in enumerate(A):
if x % 2 == 0:
ans[t] = x
t += 2
t = 1
for x in enumerate(A):
if x % 2 == 1:
ans[t] = x
t += 2
return ans