算法现在就是大厂、外企的硬指标。开发、测开、测试,想往上总是绕不开的。
这是今天的每日一题,看到是个简单的,就做了下(非官方题解)。
题目描述
难度:【简单】 标签:【数组】、【排序】
给你一个整数数组 nums ,其中总是存在 唯一的 一个最大整数 。 请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍 。 如果是,则返回 最大元素的下标 ,否则返回 -1 。
题目地址:https://leetcode-cn.com/problems/largest-number-at-least-twice-of-others/
示例
示例 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 至少是其他数字的两倍。
解题
因为总是存在 唯一的 一个最大整数
,于是我首先找出数组里的最大的数,记录下来这个数。
然后遍历数组,逐个比较。
至少是数组中每个其他数字的两倍
,就是说当2 * i > max_num
,就返回 -1 结束。
后来提交报错,发现忽略一个问题,就是当遍历到最大数本身的时候是满足上述的比较条件的,返回 -1 。
所以我增加了一步,不仅记录下最大的数,还记录了最大数的下标,然后 remove 了这个最大数,接着遍历即可。
全部遍历完成没有中途返回 -1,则返回最大数的下标。
class Solution: def dominantIndex(self, nums: List[int]) -> int: max_num = max(nums) index_max = nums.index(max_num) nums.remove(max_num) for i in nums: if 2 * i > max_num: return -1 return index_max