Leetcode 922 Sort Araay By Parity II

发表于 |

给定一个非负整数数组 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] 也会被接受。

提示：

  1. 2 <= A.length <= 20000
  2. A.length % 2 == 0
  3. 0 <= A[i] <= 1000

题意分析：
给定一个数组，变动其内的若干数使得偶数位上的数是偶数，奇数位上的数是奇数。

思路分析：
当时周测的时候因为争分夺秒，我就直接写的O(n) space的方法
但实际上这道题是可以用O(1)的方法来求解的

因为题目中说到A中奇数偶数各一半，那么当我找到一个奇数位上的偶数，就一定存在某个奇数存在于偶数位置上。
我们可以使用双指针法来求解，一个指针专门记录奇数位上的偶数，另一个则记录偶数位上的奇数

1
2
3
4
5
6
7
8
9
10
11
# 164ms
class Solution(object):
    def sortArrayByParityII(self, A):
        n = len(A)
        l,r = 0,n-1
        while l < n and r >= 0:
            while l < n and A[l] % 2 == 0: l += 2
            while r >=0 and A[r] % 2 == 1: r -= 2
            if l < n and r >= 0:
                A[l], A[r] = A[r], A[l]
        return A