[LeetCode]324. 摆动排序 II(Wiggle Sort II)

题目描述

给定一个无序的数组 nums,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]… 的顺序。

示例 1:

输入: nums = [1, 5, 1, 1, 6, 4]
输出: 一个可能的答案是 [1, 4, 1, 5, 1, 6]

示例 2:

输入: nums = [1, 3, 2, 2, 3, 1]
输出: 一个可能的答案是 [2, 3, 1, 3, 1, 2]
说明:
你可以假设所有输入都会得到有效的结果。

进阶:
你能用 O(n) 时间复杂度和 / 或原地 O(1) 额外空间来实现吗?

解题思路

将原始数组排序,分为小数部分和大数部分,再将其穿插即可。
注:如果直接将大数部分穿插到奇数索引位置,则会出现问题:nums = [1, 2, 4, 4, 4, 6] 结果为 [1, 4, 2, 4, 4, 6],不满足要求,但若降序穿插即可满足要求:[4, 6, 4, 5, 1, 4],因为这使得中间重复的数字相差一位。

代码

Python 3.6

1
2
3
4
5
6
7
8
9
class Solution:
def wiggleSort(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
nums.sort(reverse=True)
odd = nums[:len(nums) // 2]
even = nums[len(nums) // 2:]
nums[::2], nums[1::2] = even, odd

执行用时: 140 ms
内存消耗: 14.2 MB