找出数组中第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
上一篇: 计算1到10的阶乘之和
下一篇: 有向图中以一个顶点为起始点的所有路径