空でない配列を指定すると、この配列で3番目に大きい数を返します。存在しない場合は、配列内の最大数が返されます。アルゴリズムの時間計算量はO(n)でなければなりません。
例1:
入力:[3、2、1]
出力:1
説明:3番目に大きい数は1です。
例2:
入力:[1、2]
出力:2
説明:3番目に大きい数値が存在しないため、最大の数値2を返します。
例3:
入力:[2、2、3、1]
出力:1
説明:3番目に大きい一意の番号を参照する3番目に大きい番号を返す必要があることに注意してください。
値が2の数値が2つあり、どちらも2番目にランク付けされています。
出典:LeetCode
リンク:https://leetcode-cn.com/problems/third-maximum-number
方法1:1回トラバースする
public static int thirdMax2(int[] nums) {
if (nums.length == 1) {
return nums[0];
}
if (nums.length == 2) {
return Math.max(nums[0], nums[1]);
}
int max1 = Integer.MIN_VALUE;
int max2 = Integer.MIN_VALUE;
int max3 = Integer.MIN_VALUE;
int flag = 0;
boolean f = true;
for (int i = 0; i < nums.length; i++) {
if (nums[i] > max1) {
flag++;
//原先第二大传递给第三大
max3 = max2;
//原先最大值传递给第二大
max2 = max1;
//更新最大值
max1 = nums[i];
} else if (nums[i] > max2 && nums[i] < max1) {
flag++;
max3 = max2;
max2 = nums[i];
} else if (nums[i] > max3 && nums[i] < max2) {
flag++;
max3 = nums[i];
} else if (nums[i] == Integer.MIN_VALUE && f) {
flag++;
f = false;
}
}
return flag >= 3 ? max3 : max1;
}
方法2:時間計算量がO(n)として指定されていない場合に使用可能
public static int thirdMax2(int[] nums) {
HashSet<Integer> set = new HashSet<>();
for (int num : nums) {
set.add(num);
}
Object[] objects = set.toArray();
Arrays.sort(objects);//O(nlogn)
int thirdmax;
if (set.size() < 3) {
thirdmax = (int) objects[objects.length - 1];
} else {
thirdmax = (int) objects[objects.length - 3];
}
return thirdmax;
}