寻找最大值下标

题目描述:

在本题中,有一个数组 arr,在这个数组中,除了一个最大值之外,其它所有的值都相等

现在你需要在不直接进行比较值的方法,来查找这个最大值所在的下标。你需要借助题目提供的 compare()函数来帮助判断。
该函数定义如下: compare() 函数会接收 4 个参数,假定为 a,b,c,d,其中 a <= b 并且 c <=
d。该函数可以判断数组 arr 中闭区间 [a, b] 的和与闭区间 [c, d] 之间的和之间的大小,函数的返回值为 int
类型,具体返回值含义如下:

  • 1:闭区间 [a, b] 的和更大
  • -1:闭区间 [c, d] 的和更大
  • 0:两个区间的和相等
    请你返回最大值的下标,你可以假定最大值一定存在。

我的解题思路:List 可以直接使用 max() 获取最大值,然后循环遍历找到最大值的下标即可。虽然解决了问题,但是没有按照题目的要求。时间复杂度:O(n),空间复杂度:O(1)。

from typing import (
    List,
)

class Solution:
    """
    @param arr: a list
    @param helper: You can using compare function by helper.compare(int, int, int, int)
    @return: max element index
    """
    def get_max_element_index(self, arr: List[int], helper: Helper) -> int:
        # write your code here
        
        target = max(arr)
        
        for i in range(len(arr)):
            if arr[i] == target:
                return i
                break

看了大家的题解,发现大家都用的类似二分法,同时看前一半和后一半,哪边更大,最大值就在哪边。以此类推直到剩下一个数,返回。
代码如下:

from typing import (
    List,
)

class Solution:
    """
    @param arr: a list
    @param helper: You can using compare function by helper.compare(int, int, int, int)
    @return: max element index
    """
    def get_max_element_index(self, arr: List[int], helper: Helper) -> int:
        # write your code here
        # 进行判空操作
        if not arr: return -1
        # 类似二分,前看前一半和后一半,哪边更大,最大值就在哪边。继续循环直到剩下一个数,返回。
        left, right = 0, len(arr) - 1

        while (left < right):
            l_end = (left + right) // 2
            r_st = (left + right + 1) // 2
            if (helper.compare(left, l_end, r_st, right) == 1): #左边更大
                right = l_end
            else:
                left = r_st
        return left
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值