给定一个非空数组,返回此数组中第三大的数。如果不存在,则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
示例一:
输入: [1, 2]
输出: 2
解释: 第三大的数不存在, 所以返回最大的数 2 .
示例二:
输入: [2, 2, 3, 1]
输出: 1
解释: 注意,要求返回第三大的数,是指第三大且唯一出现的数。存在两个值为2的数,它们都排第二。
注意点:最开始定义two和three时,不能写成 int two = Integer.MIN_VALUE;
因为 Integer.MIN_VALUE 的值为 -2147483648 ,同样可能是这个数组中的数。
所以要选择取值范围大于 Integer 的 Long 的最小值。也就是 long two = Long.MIN_VALUE;
思路:
- 用三个变量来存放第一大,第二大,第三大的元素的变量,分别为one, two, three;
- 遍历数组,若该元素比 one 大则将它们三个都往后移动一个格,并将此时的元素赋值给 one;若比 two 大(比one小)则 two 和 three 往后移动一格;若比three大(比one和two小)则赋值给 three。
- 最后得到第三大的元素,若没有则返回第一大的元素。
public class LC414 {
public int thirdMax(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int one = nums[0];
long two = Long.MIN_VALUE;
long three = Long.MIN_VALUE;
for (int n : nums) {
if (n == one || n == two || n == three) {
continue; // 如果存在过,就跳过
}
if (n > one) {
three = two;
two = one;
one = n;
} else if (n > two) {
three = two;
two = n;
} else if (n > three) {
three = n;
}
}
return three == Long.MIN_VALUE ? one : (int) three; // 没有第三大的元素,就返回最大值
}
}