该题使用快速排序(优化后的逻辑)来实现了去排序查找的功能。
快排的思想,用一个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;//返回枢纽所在的位置
}
};