Peaks and Valleys:数组中的“峰”表示该元素大于等于两侧的元素,“谷”表示该元素小于等于两侧元素,实现一个算法,将数组按照峰谷的顺序进行排序。
假设数组是有序的,例如:
0 1 4 7 8 9
,通过交换0
和1
,可得
1 0 4 7 8 9
,通过交换7
和4
,可得
1 0 7 4 8 9
1 0 7 4 9 8
这样就得到了最终的结果。
class Solution {
public:
void wiggleSort(vector<int>& nums) {
sort(nums.begin(), nums.end());
int tmp;
for(size_t i = 1; i < nums.size(); i += 2)
{
tmp = nums[i];
nums[i] = nums[i - 1];
nums[i - 1] = tmp;
}
}
};
其实也可以不用排序,直接在每个谷或者峰的位置,同左边一个和右边一个进行比较,找出一个最小的或者最大的放在中间的位置即可。
class Solution {
public:
void wiggleSort(vector<int>& nums) {
int tmp;
for(size_t i = 1; i < nums.size(); i += 2)
{
size_t idx = smallIdx(nums, i - 1, i, i + 1);
if(idx != i){
tmp = nums[i];
nums[i] = nums[idx];
nums[idx] = tmp;
}
}
}
private:
size_t smallIdx(const vector<int> &nums, size_t left, size_t mid, size_t right)
{
size_t ret = nums[left] < nums[mid] ? left : mid;
if(right < nums.size()){
ret = nums[ret] < nums[right] ? ret : right;
}
return ret;
}
};