75 lines
1.7 KiB
Java
75 lines
1.7 KiB
Java
|
//给你一个非空数组，返回此数组中 第三大的数 。如果不存在，则返回数组中最大的数。
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 示例 1：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//输入：[3, 2, 1]
|
|||
|
|
//输出：1
|
|||
|
|
//解释：第三大的数是 1 。
|
|||
|
|
//
|
|||
|
|
// 示例 2：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//输入：[1, 2]
|
|||
|
|
//输出：2
|
|||
|
|
//解释：第三大的数不存在, 所以返回最大的数 2 。
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 示例 3：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//输入：[2, 2, 3, 1]
|
|||
|
|
//输出：1
|
|||
|
|
//解释：注意，要求返回第三大的数，是指在所有不同数字中排第三大的数。
|
|||
|
|
//此例中存在两个值为 2 的数，它们都排第二。在所有不同数字中排第三大的数为 1 。
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 提示：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 1 <= nums.length <= 104
|
|||
|
|
// -231 <= nums[i] <= 231 - 1
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 进阶：你能设计一个时间复杂度 O(n) 的解决方案吗？
|
|||
|
|
// Related Topics 数组 排序
|
|||
|
|
// 👍 230 👎 0
|
|||
|
|
|
|||
|
|
package leetcode.editor.cn;
|
|||
|
|
|
|||
|
|
import java.util.Arrays;
|
|||
|
|
import java.util.TreeSet;
|
|||
|
|
|
|||
|
|
//414:第三大的数
|
|||
|
|
class ThirdMaximumNumber {
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
//测试代码
|
|||
|
|
Solution solution = new ThirdMaximumNumber().new Solution();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//力扣代码
|
|||
|
|
//leetcode submit region begin(Prohibit modification and deletion)
|
|||
|
|
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();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//leetcode submit region end(Prohibit modification and deletion)
|
|||
|
|
|
|||
|
|
}
|