关于了leetcode的又一个小练习

寻找数组中主导索引

最新推荐文章于 2020-05-01 15:31:03 发布

原创最新推荐文章于 2020-05-01 15:31:03 发布 · 202 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于找出数组中最大值及其位置，并判断该最大值是否至少为次大值的两倍。通过一次遍历实现，提高了效率。

class Solution:
    def dominantIndex(self, nums):
        m = 0
        n = 0
        pos = 0
        i = 0
        for num in nums:
            if(num>m):
        	    pos = i
        	    n = m
        	    m = num
            elif(num > n):
                n = num
            i = i+1   	
        if(m >= n*2):
            return pos
        else: 
            return -1

大概的思路是，找到最大的元素，然后找到第二大的元素，比较第二大的元素是不是小于最大的1/2，或者比较最大的元素是不是第二大的两倍也可以。那么怎么找到第二大的元素呢（最大那个比较容易，我就不说了），第二大的当然可以再弄一个循环找出来，也可以。或者，思考第二大元素的来源，要么是最大那个元素的上一个取值，要么是取到最大元素后，后面的一个较大值，如此加入相应判断条件即可，其他的都是需要什么就加上什么，比如为了返回位置，我特地添加了pos记录最大取值时候的位置。如此。