Leetcode --- 414. 第三大的数（数组）

414. 第三大的数

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

输入: [3, 2, 1]
输出: 1
解释: 第三大的数是 1.

示例2：

输入: [1, 2]
输出: 2
解释: 第三大的数不存在, 所以返回最大的数 2 .

示例3：

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

JAVA

思路1：

1. 使用set集合去重，使用sort排序。

class Solution {    
   public int thirdMax(int[] nums) {        
      int res = 0;        
      Set <Integer> set=new HashSet<Integer>();        
      for (int i=0;i<nums.length;i++){            
           set.add(nums[i]);        
      }        
      int[] arr = new int[set.size()];        
      int index=0;        
      for(int i:set){            
          arr[index++]=i;        
      }        
      Arrays.sort(arr);        
      if (arr.length<3){          
          res= arr[arr.length-1];        
      }else{        
          res=arr[arr.length-3];        
      }       
      return res;    
   }
 }

思路2：

1. 使用TreeSet既可以排序又可以去重

class Solution {
    public int thirdMax(int[] nums) {
        TreeSet<Integer> retset = new TreeSet<>();
        for(int i=0;i<nums.length;i++){
            retset.add(nums[i]);
            if(retset.size()>3) retset.pollFirst();
        }if(retset.size()==3) return retset.first();
        return retset.last();
    }
}