最简单的方法就是排序然后输出,但是首先要用set去重
class Solution {
public:
int thirdMax(vector<int> &nums) {
set<int> s;
for (int i = 0; i < nums.size(); i++) {
s.insert(nums[i]);
if (s.size() > 3) {//维护一个空间为3的set
s.erase(s.begin());
}
}
if (s.size() == 3)//如果set空间为3输出第三大的数
return *s.begin();
return *s.rbegin();//否则输出最大的数,这里不能用s.end()-1.因为set的迭代器是不连续的不支持operator -。
}
};
第二种方法就是遍历一遍,用三个数依次表示最大的数,第二大的数,第三大的数,这样的实录可以以此类推。
class Solution {
public:
int thirdMax(vector<int>& nums) {
long a=INT64_MIN,b=INT64_MIN, c=INT64_MIN;
for (int num : nums){
if (num > a){//保存最大的数
c = b;//注意顺序
b = a;
a = num;
}
else if (num > b && num < a){//保存第二大的数
c = b;
b = num;
}
else if (num < b && num > c){//保存第三大的数
c = num;
}
}
return (c == INT64_MIN) ? a : c;//INT64_MIN注意测试用例[1,2,-2147483648]
}
};