280. Wiggle Sort
class Solution {
public:
void wiggleSort(vector<int>& nums) {
for (int i = 1; i < nums.size(); i++)
if ((i & 1) == (nums[i - 1] > nums[i])) swap(nums[i], nums[i - 1]);
return;
}
};
why is this greedy solution can ensure previous sequences and coming sequences W.R.T position i wiggled?
My explanation is recursive,
suppose nums[0 .. i - 1] is wiggled, for position i:
if i is odd, we already have, nums[i - 2] >= nums[i - 1],
if nums[i - 1] <= nums[i], then we does not need to do anything, its already wiggled.
if nums[i - 1] > nums[i], then we swap element at i -1 and i. Due to previous wiggled elements (nums[i - 2] >= nums[i - 1]), we know after swap the sequence is ensured to be nums[i - 2] > nums[i - 1] < nums[i], which is wiggled.
similarly,
if i is even, we already have, nums[i - 2] <= nums[i - 1],
if nums[i - 1] >= nums[i], pass
if nums[i - 1] < nums[i], after swap, we are sure to have wiggled nums[i - 2] < nums[i - 1] > nums[i].
The same recursive solution applies to all the elements in the sequence, ensuring the algo success.