747. 至少是其他数字两倍的最大数(Python)

题目

难度:★☆☆☆☆
类型:数组

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

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

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

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

示例

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

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

解答

我们首先准备三个变量并初始化:
最大数(max_1):-inf
次大数(max_2):-inf
和最大数的下标(max_id):-1

在遍历过程中,我们不断更新这三个变量,在最后判断最大数与次大数的2倍之间的差,并根据判断结果返回相应参数即可。

class Solution:
    def dominantIndex(self, nums):
        if not nums:
            return -1

        max_1 = max_2 = -float('inf')
        max_id = -1
        for i in range(len(nums)):
            if nums[i] > max_1:
                max_1, max_2, max_id = nums[i], max_1, i
            elif max_2 < nums[i] < max_1:
                max_2 = nums[i]

        return max_id if max_1 >= max_2 * 2 else -1

如有疑问或建议,欢迎评论区留言~

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容