题目描述:
给定一个无序的数组 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) 额外空间来实现吗?
解题思路:
排序后,以中间为分割点,将较大和较小的数交叉存入新数组中即可。更多参考。
AC代码:
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;
}
};