题目描述:
在本题中,有一个数组 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