题目
题干
该问题 摆动排序Ⅱ,看题面:
摆动排序Ⅱ
Given an integer array nums, reorder it such that nums[0] < nums[1] > nums[2] < nums[3]…
You may assume the input array always has a valid answer.
给你一个整数数组 nums,将它重新排列成 nums[0] < nums[1] > nums[2] < nums[3]… 的顺序。
你可以假设所有输入数组都可以得到满足题目要求的结果。
示例
示例 1:
输入:nums = [1,5,1,1,6,4]
输出:[1,6,1,5,1,4]
解释:[1,4,1,5,1,6] 同样是符合题目要求的结果,可以被判题程序接受。
示例 2:
输入:nums = [1,3,2,2,3,1]
输出:[2,3,1,3,1,2]
题解
C++
class Solution {
public:
void wiggleSort(vector<int>& nums){
vector<int> temp(nums);//副本
int n = (nums.size()+1)/2, len = nums.size();
sort(temp.begin(), temp.end());//排序
reverse(temp.begin(),temp.begin()+n);//前半段逆序
reverse(temp.begin()+n,temp.end());//后半段逆序
int i = 0, j = n, k = 0;
while(k < len)
{
nums[k++] = temp[i++];//第一段的一个数据
if(k < len)//防止奇数情况进来越界
nums[k++] = temp[j++];//第二段的一个数据
}
}
};
Python
- 不反转
class Solution(object):
def __init__(self, nums: list):
self.nums = nums
def wiggle_sort(self):
nums_len = len(self.nums)
j = (nums_len + 1) // 2
temp = sorted(self.nums)
k = i = 0
while k < nums_len:
self.nums[k] = temp[i]
k += 1
i += 1
if k < nums_len: # 防止奇数情况进来越界
self.nums[k] = temp[j] # 第二段的一个数据
k += 1
j += 1
return self.nums
- 反转
class Solution(object):
def __init__(self, nums: list):
self.nums = nums
def wiggle_sort(self):
nums_len = len(self.nums)
# 前半段逆序,后半段逆序
n = (nums_len + 1) // 2
nums_sort = sorted(self.nums)
temp = sorted(nums_sort[:n], reverse=True)
temp.extend(sorted(nums_sort[n:], reverse=True))
k = i = 0
j = n
while k < nums_len:
self.nums[k] = temp[i]
k += 1
i += 1
if k < nums_len: # 防止奇数情况进来越界
self.nums[k] = temp[j] # 第二段的一个数据
k += 1
j += 1
return self.nums