题目地址: https://leetcode.com/problems/sort-array-by-parity-ii

题目描述

Given an array A of non-negative integers, half of the integers in A are odd, and half of the integers are even.

Sort the array so that whenever A[i] is odd, i is odd; and whenever A[i] is even, i is even.

Youmay return any answer array that satisfies this condition.

Example 1:

Input: [4,2,5,7]
Output: [4,5,2,7]
Explanation: [4,7,2,5], [2,5,4,7], [2,7,4,5] would also have been accepted.

Note:

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

题目大意

把一个数组重新排序,使得偶数位置全是偶数,奇数位置全是奇数。

解题方法

使用奇偶数组

直接使用两个数组分别存放奇数和偶数,然后结果就是在这两个里面来回的选取就好了。这种做法比较简单,打比赛比较适用。

时间复杂度是O(N),空间复杂度是O(N)。

class Solution(object):
    def sortArrayByParityII(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        odd = [x for x in A if x % 2 == 1]
        even = [x for x in A if x % 2 == 0]
        res = []
        iseven = True
        while odd or even:
            if iseven:
                res.append(even.pop())
            else:
                res.append(odd.pop())
            iseven = not iseven
        return res

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

排序

先对A进行排序,使得偶数都在前面,奇数都在后面,然后每次从前从后各取一个数,然后放到结果里就好了。

class Solution:
    def sortArrayByParityII(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        A.sort(key = lambda x : x % 2)
        N = len(A)
        res = []
        for i in range(N // 2):
            res.append(A[i])
            res.append(A[N - 1 - i])
        return res

1 2 3 4 5 6 7 8 9 10 11 12 13

时间复杂度是O(NlogN),空间复杂度是O(1)。

奇偶数位置变量

先把结果数组创建好,然后使用奇偶数两个变量保存位置,然后判断A中的每个数字是奇数还是偶数,对应放到奇偶位置就行了。

时间复杂度是O(N),空间复杂度是O(1)。

class Solution:
    def sortArrayByParityII(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        N = len(A)
        res = [0] * N
        even, odd = 0, 1
        for a in A:
            if a % 2 == 1:
                res[odd] = a
                odd += 2
            else:
                res[even] = a
                even += 2
        return res

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

时间复杂度是O(N),空间复杂度是O(N)。

DDKK.COM 弟弟快看-教程,程序员编程资料站,版权归原作者所有

本文经作者:负雪明烛 授权发布,任何组织或个人未经作者授权不得转发