在一个给定的数组nums中,总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是,则返回最大元素的索引,否则返回-1。
示例 1:
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.
示例 2:
输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.
提示:
nums 的长度范围在[1, 50].
每个 nums[i] 的整数范围在 [0, 99].
解:
找出至少是其他数字两倍的最大数的索引,意味着只需找出最大值和次大值进行比较即可。
这里使用选择排序,因为选择排序的基本思想是:
一组记录经过第一轮比较后(0~n-1)把最小(大)元素放在首位,第二轮比较(1 ~ n-1)把次小(大)放在第二位……依此类推直至排序完成。
这里只需最大值和次大值,两轮比较即可解决问题
代码:
class Solution {
public int dominantIndex(int[] nums) {
if (nums.length == 1)
return 0;
int res = 0; //最大值索引
//只需进行选择排序两轮比较即可
for (int i = 0; i < 2; i++)
{
int index = i;
for (int j = i + 1; j < nums.length; j++)
{
if (nums[index] < nums[j])
{
index = j;
}
}
if (index != i){
swap(nums, i, index);
}
//记录最大值的索引
if (i == 0)
res = index;
}
return nums[0] >= 2*nums[1] ? res : -1;
}
private void swap(int[] nums, int a, int b)
{
int tmp = nums[a];
nums[a] = nums[b];
nums[b] = tmp;
}
}