C++ 中的摆动排序
假设我们有一个名为 nums 的未排序数组,我们必须对其进行原地重新排序,使得 nums[0] <= nums[1] >= nums[2] <= nums[3].... 等等。
因此,如果输入像 nums = [3,5,2,1,6,4],那么输出将是 [3,5,1,6,2,4],可能还有其他答案。
为了解决这个问题,我们将按照以下步骤进行操作:
n := nums 的大小
初始化 i := 0,当 i < n - 1 时,更新 i := i + 1,执行以下操作:
如果 i 为偶数且 nums[i] > nums[i+1] 为 true,或 i 为奇数且 nums[i] > nums[i+1] 为 false,则
交换 (nums[i], nums[i + 1])
示例
让我们看一下以下实现,以获得更好的理解:
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto< v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: void wiggleSort(vector<int<& nums) { int n = nums.size(); for(int i = 0; i < n - 1; i+=1){ if((i % 2 == 0) == ( nums[i] > nums[i + 1])){ swap(nums[i], nums[i + 1]); } } } }; main(){ vector<int< v = {3,5,2,1,6,4}; Solution ob; ob.wiggleSort(v); print_vector(v); }
输入
{3,5,2,1,6,4}
输出
[3, 5, 1, 6, 2, 4, ]
广告