lintcode----前k大的数

题目描述;
在一个数组中找到前K大的数

样例:
给出 [3,10,1000,-99,4,100], k = 3.
返回 [1000, 100, 10]

思路讲解:这里我的第一想法就是利用一个小的res,首先将里面添加进nums的前k个数,然后对其排序,之后我们就将nums位置k后面的数逐个的与res数组比较这里就用到了对有序数组的插入排序,结果这样很不辛的在70%的时候超时了,后面我看了一下标签,发现可以用优先队列,首先我们将所有的数,都进入优先队列,然后将其前k拿出来就好了。后面想了一下,我之前的那种算法适用的场景是k很小,并且数据很大的时候。

代码详解:

class Solution {
public:
    /**
     * @param nums: an integer array
     * @param k: An integer
     * @return: the top k largest numbers in array
     */
    vector<int> topk(vector<int> &nums, int k) {
        // write your code here
        int length=nums.size();

        priority_queue<int> q;

        vector<int>res;

        for(int i=0;i<length;i++){
            q.push(nums[i]);
        }

        int count=0;
        while(count!=k){
            res.push_back(q.top());
            q.pop();
            count++;
        }

        printvector(res);
        return res;

    }

    void printvector(vector<int>res){
        //cout<<"length is "<<res.size()<<endl;
        for(int i=0;i<res.size();i++){
            cout<<res[i]<<" ";
        }
        cout<<endl;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_34355232/article/details/79530677