查找第K大的元素

查找第K大的元素

http://www.nowcoder.com/questionTerminal/673454422d1b4a168aed31e449d87c00

题解

该问题为排序问题,给出几种常用方法:

方法(一)

使用vector,用库函数sort进行排序。

#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
bool compare(const int &a,const int &b){
       return a>b;
   }
int main(){
   int num; 
   vector<int> v; 
//将字符串中的数字放入vector中 
  while(getchar()!=']')
    {
        cin>>num;
        v.push_back(num);   
    }
//sort排序,默认为由小到大,需要由大到小自己写compare函数 
   sort(v.begin(),v.end(),compare);
//输出第三大的数字
   cout<<v[2];
   return 0;
}

方法(二)

冒泡排序

#include<vector>
#include<iostream>
using namespace std;
int main(){
   int num; 
   vector<int> v; 
//将字符串中的数字放入vector中 
  while(getchar()!=']')
    {
        cin>>num;
        v.push_back(num);   
    }
//冒泡排序 
  for(int i = 0 ;i<v.size();i++){
        //第i趟比较
        for(int j = 0 ;j<v.size()-i-1;j++){
            //开始进行比较,如果arr[j]比arr[j+1]的值小,那就交换位置
            if(v[j]<v[j+1]){
                int temp=v[j];
                v[j]=v[j+1];
                v[j+1]=temp;
            }
        }
    }

//输出第三大的数字
   cout<<v[2];
   return 0;
}

方法(三)

快速排序

#include<vector>
#include<iostream>
using namespace std;
//快排
    void swap(vector<int> &arr, int i, int j) {
        if (arr.size()==0 || i >= arr.size() || j >= arr.size() || i < 0 || j < 0) {
            return;
        }
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
    int partion(vector<int> &data, int begin, int end) {
        int temp = data[begin];
        while (begin < end) {
            while (begin < end && data[end] <= temp) {
                end--;
            }
            swap(data, begin, end);
            while (begin < end && data[begin] > temp) {
                begin++;
            }
            swap(data, begin, end);
        }
        return begin;
    }
    int findKthNum(vector<int> &data, int k) {
        int begin = 0, end = data.size() - 1;
        int pos = 0;
        while (begin <= end) {
            pos = partion(data, begin, end);
            if (pos == k - 1) {
                return data[pos];
            } else if (pos > k - 1) {
                end = pos - 1;
            } else {
                begin = pos + 1;
            }
        }
        return -1;
    } 
int main(){
   int num; 
   vector<int> v; 
//将字符串中的数字放入vector中 
  while(getchar()!=']')
    {
        cin>>num;
        v.push_back(num);   
    }
//调用快排输出第三大的数 
   cout<<findKthNum(v,3);
   return 0;
}

方法(四)

求第n个最大的数先建立最小堆把海量数据向最小堆添加n个数,堆的内部会自动排序,不用管它的内部是什么操作。然后我们往最小堆进行插入操作,因为最小堆的特点就是堆顶元素最小。若插入元素小于堆顶元素 ,则它小于堆中任何一个元素, 所以摒弃该元素遍历下一个元素。若插入元素大于堆顶元素则进行插入, 插入后堆内自动排序,堆顶元素又成为该堆的最小元素 ,原堆顶元素出堆。持续遍历完得到的就是所有元素中第n个最大数。

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
int main(){
    //在最短时间内找到所有整数中最大/最小的n个数并且打印;
    //找top最大的用小根堆;找top最小的用大根堆
    //时间复杂度 O(n)*log2 n
    vector<int> v;//保存字符串中所有出现的数 
    int n=0;
    int num;
    while(getchar()!=']') {
        cin>>num;
        n++;
        v.push_back(num);   
    }
    priority_queue<int, vector<int>, greater<int> > minHeap; //建立最小堆
    //注意:最小堆要加上greater<int>
    int j = 3;
    for (int i = 0; i < j; i++)    {
        minHeap.push(v[i]);
    }
    for (int i = j; i < n; ++i){
        //遍历判断容器中剩余的每个数与堆顶元素的大小
        //若大于堆顶元素 则把堆顶元素出堆 把该元素入堆
        if (v[i] > minHeap.top())
        {
            minHeap.pop();
            minHeap.push(v[i]);
        }
    }
        cout << minHeap.top() << " ";
    cout << endl;
    return 0;
}
全部评论

相关推荐

拒绝无效加班的小师弟很中意你:求职意向没有,年龄、课程冗余信息可以删掉,需要提升项目经历。排版需要修改。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务