力扣刷题(347. 前 K 个高频元素)快速排序

该题使用快速排序(优化后的逻辑)来实现了去排序查找的功能。

快排的思想,用一个while循环每次去在区间中寻找枢纽值,然后将枢纽值左边的区间调用递归,改变low值为pivot+1,进行下一次循环,知道low>=high.

寻找pivot:采用三值取中的思想,取左右边界和中间的三个数中的中间值作为枢纽值,将枢纽值放到数组的左边界,然后用于一个临时变量来存储枢纽值,下面就是一个while循环,不断的取更改low和high值,最后将low位置的值赋值为pivot值,保证枢纽值左边的为小于枢纽值,右边的大于枢纽值。最后返回枢纽位置。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> check;
        for(auto &k:nums)++check[k];//初始化哈希表
        vector<pair<int,int>> vals(check.begin(),check.end());//利用哈希表初始化数组
        quick_sort(vals,0,vals.size()-1,k-1);
        vector<int>ans(k,0);
        for(int i=0;i<k;++i)ans[i]=vals[i].first;
        return ans;
    }

    void quick_sort(vector<pair<int,int>>& vals,int low,int high,int target)
    {  
        while(low<high)//这里一个while循环是为了不断对右边部分进行快速排序
        {
            int pivot=find_pivot(vals,low,high);//找到枢纽并将数组一分为二
            quick_sort(vals,low,pivot-1,target);
            if(pivot>target)break;//我只需要对前target进行排序,所以大于的部分可以不管
            low=pivot+1;//下一次循环的时候find过程其实就是对右半边进行排序,递归的过程相对消耗资源
        }
    }

    int find_pivot(vector<pair<int,int>>& vals, int low,int high)//寻找枢纽
    {
        int mid=low+(high-low)/2;
        if(vals[low].second<vals[high].second)//左边不小于右边
            swap(vals[low],vals[high]);
        if(vals[mid].second<vals[high].second)//中间不小于右边
            swap(vals[mid],vals[high]);
        if(vals[mid].second<vals[low].second)//保证左边是处于中间大小的数值
            swap(vals[low],vals[mid]);
        auto pivot=vals[low];//找一个临时的变量存储pivot值
        while(low<high)
        {
            while(low<high&&pivot.second>=vals[high].second)
                --high;
            vals[low]=vals[high];
            while(low<high&&pivot.second<=vals[low].second)
                ++low;
            vals[high]=vals[low];
        }
        vals[low]=pivot;//最后low所在的位置就是应该防止枢纽的位置
        return low;//返回枢纽所在的位置
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

yanzhe1

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值