原题地址:https://leetcode-cn.com/problems/wiggle-sort-ii/submissions/
题目描述:
给定一个无序的数组 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) 额外空间来实现吗?
解题方案:
排序问题,先使用sort()函数进行排序,再对数据重新定位。
代码:
class Solution {
public:
void wiggleSort(vector<int>& nums) {
if (nums.size() <= 1)
return;
sort(nums.begin(), nums.end());
int len = nums.size(), k = 1, high = (len % 2) ? len - 1 : len - 2, mid = nums[len / 2];
vector<int> ans(len, mid);
for (int i = len - 1; i >= 0 && nums[i] > mid; --i, k += 2)
ans[k] = nums[i];
for (int i = 0; i < len && nums[i] < mid; ++i, high -= 2)
ans[high] = nums[i];
nums = ans;
}
};