程序员面试金典 10.11

Peaks and Valleys:数组中的“峰”表示该元素大于等于两侧的元素,“谷”表示该元素小于等于两侧元素,实现一个算法,将数组按照峰谷的顺序进行排序。

假设数组是有序的,例如:
0 1 4 7 8 9,通过交换01,可得
1 0 4 7 8 9,通过交换74,可得
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;
    }
};
发布了194 篇原创文章 · 获赞 2 · 访问量 7701

猜你喜欢

转载自blog.csdn.net/RayoNicks/article/details/105600033