LeetCode 280. Wiggle Sort

最新推荐文章于 2024-11-04 19:20:48 发布

原创最新推荐文章于 2024-11-04 19:20:48 发布 · 198 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 题解专栏收录该内容

288 篇文章

订阅专栏

本文介绍了一种数组摆动排序算法的两种实现方法。一种是先排序数组再进行局部交换，另一种是在原数组上直接通过比较和交换元素来达到摆动排序的要求。这两种方法都能使数组元素满足奇数位置上的元素大于两边偶数位置上的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

I didn't come up the second method. I guess in limited time interview, it is quite hard to do so...

#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;

/*
  Given an unsorted array nums, reorder it in-place that
  nums[0] <= nums[1] >= nums[2] <= nums[3] >= nums[4] <= nums[5] >= nums[6]
  For example:
  Given nums = [3, 5, 2, 1, 6, 4], one possible answer is [1, 6, 2, 5, 3, 4]
*/
// Time complexity(NlgN)
void wiggleSort(vector<int>& nums) {
  sort(nums.begin(), nums.end());
  for(int i = 1; i < nums.size() - 1; i += 2) {
    swap(nums[i], nums[i + 1]);
  }
}

// what if we do not sort first.
void wiggleSortII(vector<int>& nums) {
  for(int i = 1; i < nums.size(); ++i) {
    if((i%2 == 1 && nums[i] < nums[i-1]) || (i%2 == 0 && nums[i] > nums[i-1])) {
      swap(nums[i], nums[i-1]);
    }
  }
}

int main(void) {
  vector<int> nums{3, 5, 2, 1, 6, 4};
  wiggleSortII(nums);
  for(int i = 0; i < nums.size(); ++i) {
    cout << nums[i] << " ";
  }
  cout << endl;
}