[Leetcode] 第324题 摆动排序II

Posted 2021-01-31 龙场驿

一、题目描述

给定一个无序的数组 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) 额外空间来实现吗？

二、题目解析

1）首先对无序数组进行排序，使用额外的空间tmp
2）排序之后，进行取数，从中间分成左右两部分，分别从两部分的末尾开始取
3）这样的时间复杂度是n*logn，空间复杂度是n

三、代码实现

 1 class Solution {
 2 public:
 3     void wiggleSort(vector<int>& nums) {
 4         int n = nums.size();
 5         if (n < 2)return;
 6         vector<int>tmp = nums;
 7         sort(tmp.begin(), tmp.end());
 8         int k = (n + 1) / 2, t = n;//如果是奇数，那么中间的数应该是属于前半部分的
 9         for (int i = 0; i < n; ++i) {
10             nums[i] = i % 2 ? tmp[--t] : tmp[--k];
11         }
12     }
13 };