函数在接受用户输入时不返回索引

Question

函数应返回任何有效峰值元素的索引，但在接受用户输入时不返回索引。

class Solution:   
def __init__(self, arr):
self.arr = arr
def peakElement(self):
max=self.arr[0]
for x in self.arr:
if x>max:
max=x

return self.arr.index(max)        

p=Solution(list(input()))
print(p.peakElement())

Answer 1

您可以遍历列表并过滤掉每个元素，其中的值等于max()。self.arr.index(max)将只返回第一个索引。

class Solution:
def __init__(self, arr):
self.arr = arr
def peakElement(self):
maximum = max(self.arr)
return [i for i, j in enumerate(self.arr) if j == maximum]

p = Solution(list(input()))
print(p.peakElement())

输入/输出：

['5', '6', '5', '6']
[1, 3]