题目描述
给你一个整数数组 nums ,其中总是存在 唯一的 一个最大整数 。
请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍 。如果是,则返回 最大元素的下标 ,否则返回 -1 。
示例 1:
输入:nums = [3,6,1,0]
输出:1
解释:6 是最大的整数,对于数组中的其他整数,6 大于数组中其他元素的两倍。6 的下标是 1 ,所以返回 1 。
示例 2:
输入:nums = [1,2,3,4]
输出:-1
解释:4 没有超过 3 的两倍大,所以返回 -1 。
示例 3:
输入:nums = [1]
输出:0
解释:因为不存在其他数字,所以认为现有数字 1 至少是其他数字的两倍。
题解思路
方法一:两遍遍历
- 第一遍确定最大数字的下标
- 第二遍确定是否至少是其他数字的两倍(是 —— 返回下标;不是 —— 返回 -1)
代码如下:
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int res= 0, sz = nums.size();
for (int i = 1; i < sz; ++i) {
if(nums[i] > nums[res]) {
res = i;
}
}
for(int i = 0; i < sz; ++i) {
if (res == i) continue;
i f(nums[res] >= nums[i]*2) {
continue;
}
else {
res = -1;
break;
}
}
return res;
}
};
方法二:一遍遍历
- 第一遍确定最大数字的下标,并且记录下第一大和第二大的数值
代码如下:
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int m = -1, n = -1, res = -1;
for(int i = 0; i < nums.size(); ++i) {
if (nums[i] > m) {
n = m;
m = nums[i];
res = i;
}
else if (nums[i] > n) {
n = nums[i];
}
}
return m >= n*2 ? res : -1;
}
};