324. Wiggle Sort II

Given an unsorted array nums, reorder it such that nums[0] < nums[1] > nums[2] < nums[3]....

Example:

(1) Given nums = [1, 5, 1, 1, 6, 4], one possible answer is [1, 4, 1, 5, 1, 6]. 
(2) Given nums = [1, 3, 2, 2, 3, 1], one possible answer is [2, 3, 1, 3, 1, 2].

Note:
You may assume all input has valid answer.

Follow Up:
Can you do it in O(n) time and/or in-place with O(1) extra space?

https://discuss.leetcode.com/topic/32929/o-n-o-1-after-median-virtual-indexing
思路:先找到中位数,然后从左往右将比中位数大的数放在奇数位置上,从右往左将比中位数小的数放在偶数位置上.

其中(1+2*i)%(n|1)这是一个设计极其巧妙的映射,这样会使得小于一半的数映射到1, 3, 5等奇数位置,也就是low会映射到的奇数位置,而大于一半的数会映射到从高到低的偶数位置,也就是high会依次映射到从(n|1)到0的偶数位置.这样就会让小于中位数的值优先从右往左放,而大于中位数的值优先从左往右放,剩下的等于中位数数的值会留在大于中位数和小于中位数的值中间.

Accessing A(0) actually accesses nums[1].
Accessing A(1) actually accesses nums[3].
Accessing A(2) actually accesses nums[5].
Accessing A(3) actually accesses nums[7].
Accessing A(4) actually accesses nums[9].
Accessing A(5) actually accesses nums[0].
Accessing A(6) actually accesses nums[2].
Accessing A(7) actually accesses nums[4].
Accessing A(8) actually accesses nums[6].
Accessing A(9) actually accesses nums[8].
class Solution {
public:
    void wiggleSort(vector<int>& nums) {
        int n = nums.size();
        nth_element(nums.begin(),nums.begin()+n/2,nums.end());
        int mid = nums[n/2];
        int i =0 , low =0, high = n-1;
        #define A(i) nums[(1+2*(i)) % (n | 1)]
        while(i <= high){
            if(A(i) > mid){
                swap(A(i++),A(low++));
            }
            else if(A(i) < mid){
                swap(A(i),A(high--));
            }
            else i++;
        }
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值