给定一个无序的数组 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) 额外空间来实现吗?
时间复杂度O(nlogn),空间复杂度O(n)
class Solution:
def wiggleSort(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
temp = sorted(nums)
if len(nums) % 2 == 0:
small, big = temp[:len(nums)//2], temp[len(nums)//2:]
else:
small, big = temp[:len(nums)//2+1], temp[len(nums)//2+1:]
left, right = len(small)-1, len(big)-1
i = 0
while left >= 0 or right >= 0:
if i % 2 == 0:
nums[i] = small[left]
left -= 1
else:
nums[i] = big[right]
right -= 1
i += 1
上面算法的简版
时间复杂度O(nlogn),空间复杂度O(1) (表面上。。)
class Solution(object):
def wiggleSort(self, nums):
nums.sort()
med = (len(nums) - 1) / 2
nums[::2], nums[1::2] = nums[med::-1], nums[:med:-1]
https://leetcode.com/problems/wiggle-sort-ii/discuss/141314/avg-time-O(n)-space-O(1)-Python-solution
网友评论