算法提高之LeetCode刷题Leetcode模拟面试

[LeetCode][Python]747. 至少是其他数字两倍

2019-08-20  本文已影响0人  bluescorpio

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

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

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

示例 1:

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

示例 2:

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

思路:
暴力型

  1. 先找到最大的数和对应的索引,删除这个最大的数,然后再剩下的列表里找第二大的数,看第一大的数是不是第二大的数的二倍。
  2. 给列表排序,看倒数第一个是不是倒数第二个的二倍以上
  3. 如果最大数必须是所有元素的至少两倍大,那么最大数一定是第二大数的至少两倍大。因此:

循环数组找出最大数和第二大数
比较两者大小

class Solution(object):
    def dominantIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) >=50 or len(nums)<1:
            return -1
        if len(nums)==1:
            return 0
        
        max_num = float('-inf')
        max_index = 0
        second_max_num = float('-inf')
        for i in range(len(nums)):
            n = nums[i]
            if n > max_num:
                second_max_num = max_num
                max_num = n
                max_index = i
            else:
                if n > second_max_num:
                    second_max_num = n
        if max_num >= 2*second_max_num:
            return max_index
        else:
            return -1

上一篇下一篇

猜你喜欢

热点阅读