要求
给你一个整数数组 nums,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]... 的顺序。
你可以假设所有输入数组都可以得到满足题目要求的结果。
示例 1:
输入:nums = [1,5,1,1,6,4]
输出:[1,6,1,5,1,4]
解释:[1,4,1,5,1,6] 同样是符合题目要求的结果,可以被判题程序接受。
示例 2:
输入:nums = [1,3,2,2,3,1]
输出:[2,3,1,3,1,2]
提示:
- 1 <= nums.length <= 5 * 104
- 0 <= nums[i] <= 5000
- 题目数据保证,对于给定的输入 nums ,总能产生满足题目要求的结果
进阶:你能用 O(n) 时间复杂度和 / 或原地 O(1) 额外空间来实现吗?
核心代码
class Solution:
def wiggleSort(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
nums.sort()
mid = len(nums[::2])
nums[::2],nums[1::2] = nums[:mid][::-1],nums[mid:][::-1]
解题思路:我们对我们的数组进行原地排序,然后我们从中间进行截断,注意一下,我们在nums[:mid][::-1],nums[mid:][::-1]都进行了从后向前的采样,是因为,我们想要将第一部分的最大值,和第二部分的最小值区分开,距离拉大,这样免得出现相等的情形,比较聪明的做法。