题目

在一个给定的数组nums中，总是存在一个最大元素 。

查找数组中的最大元素是否至少是数组中每个其他数字的两倍。

如果是，则返回最大元素的索引，否则返回-1。

示例 1:

1
2
3
4
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.

示例 2:

1
2
3
输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.

提示:

  1. nums 的长度范围在[1, 50].
  2. 每个 nums[i] 的整数范围在 [0, 99].

题解

贪心算法，只要遍历一遍，遍历第一项的时候即把它当作最大项，以后如果出现更大的，再将最大项设置为它。同时有一个flag记录是否符合大两倍的条件。

C++代码实现如下

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public:
    int dominantIndex(vector<int>& nums) {
        if (nums.size() == 0) return -1;
        if (nums.size() == 1) return 0;
        int max_index = 0;
        bool flag = true;
        for (int i=1;i<nums.size();i++){
            if (nums[max_index] > nums[i]){
                if (flag)
                    flag = nums[max_index] >= 2 * nums[i];
            }
            else{
                flag = nums[i] >= 2 * nums[max_index];
                max_index = i;
            }

        }
        if (!flag) return -1;
        else return max_index;
    }
};

Python3 实现如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution:
    def dominantIndex(self, nums: List[int]) -> int:
        if len(nums)== 0:
            return -1 
        if len(nums) == 1:
            return 0 
        max_index = 0
        flag = True
        for i in range(1,len(nums)):          
            if nums[max_index] > nums[i]:
                if flag:
                    flag = nums[max_index] >= 2 * nums[i] 
            else:
                flag = nums[i] >= 2 * nums[max_index]
                max_index = i
        if not flag:
            return -1 
        else:
            return max_index