LeetCode-Python-280. 摆动排序

给你一个无序的数组 nums, 将该数字 原地 重排后使得 nums[0] <= nums[1] >= nums[2] <= nums[3]...。

示例:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/wiggle-sort
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一种思路：

虽然题目要求原地，但是我们可以先想一个简单的非原地版本出来帮助思考。

class Solution(object):
    def wiggleSort(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        if not nums:
            return nums
        nums.sort()
        if len(nums) % 2 == 0:
            nums[:] = self.solution1(nums)
        else:
            nums[:] = self.solution2(nums)
    
    def solution1(self, nums):
        res = [nums[len(nums) // 2 - 1]]
        n = len(nums)
        flag = 1
        l1, l2 = nums[:len(nums) // 2 - 1][::-1], nums[len(nums)//2 : ]
        while len(res) < n: 
            if not flag:
                res.append(l1.pop())
                flag = 1
            else:
                res.append(l2.pop())
                flag = 0
        return res
    
    def solution2(self, nums):
        # 1 2 3 4 5
        res = [nums[len(nums) // 2]]
        n = len(nums)
        flag = 1
        l1, l2 = nums[:len(nums) // 2][::-1], nums[len(nums)//2:]
        while len(res) < n:
            if not flag:
                res.append(l1.pop())
                flag = 1
            else:
                res.append(l2.pop())
                flag = 0
        return res

第二种思路：

原地排序，线性扫描，如果跟后一项不满足对应的大小关系，就把当前元素和后一元素交换。

class Solution(object):
    def wiggleSort(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        if not nums:
            return nums
        i = 0
        while i < len(nums):
            if i % 2 == 0 and i < len(nums) - 1 and nums[i] > nums[i + 1]:
                nums[i], nums[i + 1] = nums[i + 1], nums[i]
            elif i % 2 and i < len(nums) - 1 and nums[i] < nums[i + 1]:
                nums[i], nums[i + 1] = nums[i + 1], nums[i]
            else:
                i += 1
        return nums