414. 第三大的数

class Solution {
    public int thirdMax(int[] nums) {
        TreeSet<Integer> set=new TreeSet();
        for(int i:nums){
            set.add(i);
            if(set.size()>3){
                 set.pollFirst();
            }
        }
        if(set.size()<3){
            return set.pollLast();
        }else{
            return set.pollFirst();
        }

    }
}