leetcode_324 摆动排序 II

127 阅读1分钟

要求

给你一个整数数组 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]

image.png

解题思路:我们对我们的数组进行原地排序,然后我们从中间进行截断,注意一下,我们在nums[:mid][::-1],nums[mid:][::-1]都进行了从后向前的采样,是因为,我们想要将第一部分的最大值,和第二部分的最小值区分开,距离拉大,这样免得出现相等的情形,比较聪明的做法。