414.第三大的数

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

示例 1:输入: [3, 2, 1]           输出: 1

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

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

方法一:三个变量

定义三个变量:one ,two ,three,分别用来存放数组前三个最大的值,那么一开始应该如何初始化他们呢?

int one = nums[0];
long two = Long.MIN_VALUE,three = two;

也可以将三个数均初始化为 Long.MIN_VALUE,将 two 和 three 定义为 long 类型是因为如果数组中有 Integer.MIN_VALUE(-2147483648) 这个数,会影响后续的判断,因此将它设为最小的数。

在这里不能将三个数全设为 nums[0],假如数组为 [4,3,2,1],则 one = two = three = 4,那么后面遍历数组时,本来遵守着找比 one,two 或者 three 大的数,但这样一初始化就找不到了。 

然后遍历数组,分别对三个变量进行更新。遍历数组的过程中如果当前的数与 one,two 或者 three 相等,那么就可以跳过此次判断。 最后对 three 进行判断,如果它的值没有改变,那么表示不存在第三大的数,则返回最大值 one ,否则返回所求数 three。   

public int thirdMax(int[] nums) {
        int one = nums[0];
        long two = Long.MIN_VALUE,three = two;
        for(int i = 1;i < nums.length;i++){
            int num = nums[i];
            if(num == one || num == two || num == three){
                continue;
            }
            if(num > one){
                three = two;
                two = one;
                one = num;
            }else if(num > two){
                three = two;
                two = num;
            }else if(num > three){
                three = num;
            }
        }
        return three == Long.MIN_VALUE ? one : (int)three;
}

方法二:利用 TreeSet

一、将数组排序,定义一个 TreeSet,往里面添加数字,由于 Set 的 add 方法不会添加重复元素,所以可以保证元素的唯一性,当 set 的元素小于三个就继续添加,并且记录新的元素 ret ,直到添加够 3 个。如果数组遍历结束 set 的大小没有到达 3 个,则表示不存在第三大的数字,那么返回数组的最后一个元素,否则返回 set 最后一次添加的 ret 元素。

public int thirdMax(int[] nums) {
        if(nums.length==2){
            return Math.max(nums[0],nums[1]);
        }
        Arrays.sort(nums);
        Set<Integer> set = new TreeSet<>();
        int ret = Integer.MIN_VALUE;
        for(int i = nums.length-1;i>=0;i--){
            if(!set.contains(nums[i]) && set.size()<3){
                set.add(nums[i]);
                ret = nums[i];
            }
        }
        if(set.size()==3) {
            return ret;
        }
        return nums[nums.length-1];
    }

二、TreeSet 会将添加进去的元素自动排序,那么

  1. 维护一个只有 3 个元素的 TreeSet,如果大于三个元素就把 Set 中的最小值 remove 掉。
  2. 最后如果 Set 中元素小于 3 就返回 Set 最大值(最后一个值),否则返回最小值(第一个)。
public int thirdMax(int[] nums){
       TreeSet<Integer> set = new TreeSet<>();
       for(int i = 0;i < nums.length;i++){
           set.add(nums[i]);
           if(set.size() > 3){
              set.remove(set.first());
           }
       }
       return set.size() < 3 ? set.last() : set.first();
 }

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值