欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

找出数组中第K大的数,时间复杂度O(n)

程序员文章站 2024-03-15 18:59:18
...

通常在K比较小的时候,第一思路是进行K次循环,找出第一大、第二大。。。一直到第K大。这思路当然是没有问题,但是时间复杂度较高,而且K一旦太大,就不太适用了。
可以用vector数组降序存储K个元素,符合条件的加入数组中:(K=3)

class Solution {
public:
    int thirdMax(vector<int>& nums) {
        vector<int> ans;
        for(auto n : nums){
            bool bSame=false;
            for(auto a : ans){
                if(n == a){
                    bSame=true;
                    break;
                }
            }
            if(bSame) continue;

            for(int i=0;i<ans.size();i++){
                if(n>ans[i]){
                    swap(n,ans[i]);
                }
            }

            if(ans.size()<3){
                ans.push_back(n);
            }
        }
        if(ans.empty()) return 0;
        if(ans.size()<3) return ans.front();
        return ans.back();
    }
};

1、使用 vector ans 来记录最大的三个数
2、根据题意,相同数字不占排名,所以遇到相同数字直接跳过
3、再从头比到尾,如果新数字够大,就交换
4、如果 ans 还不到 3 个,就加入
5、输出结果的时候,根据题意,有第三输出第三,没第三输出第一,如果第一也没有安全一点输出 0