【Leetcode】324. 摆动排序Ⅱ

题目描述:

给定一个无序的数组 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& 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 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;
}
};

 

你可能感兴趣的:(LeetCode)