Leetcode 第三大的数

Leetcode@第三大的数

第三大的数

给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。

示例 1:

输入: [3, 2, 1]

输出: 1

解释: 第三大的数是 1.

示例 2:

输入: [1, 2]

输出: 2

解释: 第三大的数不存在, 所以返回最大的数 2 .

示例 3:

输入: [2, 2, 3, 1]

输出: 1

解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。 存在两个值为2的数，它们都排第二。

如下为代码

class Solution {
    public int thirdMax(int[] nums) {
         HashSet<Integer> set = new HashSet<>();
        int thirdNum=0;
        for (int i = 0; i < nums.length; i++) {
            set.add(nums[i]);
        }
        if(set.size()<3){
            thirdNum = (int)Collections.max(set);
        }else {
              for (int i = 0; i <3 ; i++) {
                Integer max = Collections.max(set);
                if(i==2){
                    thirdNum=max;
                }
                set.remove(max);
            }
        }
     return thirdNum;
    }
}