Pythonでインデックスまたはローカルピークを見つけるプログラム

numsという番号のリストがあるとします。 nums内のすべてのピーク要素のインデックスを昇順で並べ替える必要があります。これら3つの条件がすべて満たされた場合のピーク要素のインデックスi：1。nums [i]とは異なる右側の次の数値が存在しないか、nums[i]よりも小さい必要があります。2。上の前の数値nums [i]とは異なる左側が存在しないか、nums[i]よりも小さい必要があります。3。左側または右側にnums[i]とは異なる数値が少なくとも1つあります。

したがって、入力がnums =[5、8、8、8、6、11、11]のような場合、8のプラトーは次のように見なされるため、出力は[1、2、3、5、6]になります。ピーク[1,2,3]。そして、11も[5、6]です。

これを解決するには、次の手順に従います-

• n：=numsのサイズ
• ans：=新しいリスト
• i：=0
• i
• i0：=i
• i
• i：=i + 1

例

理解を深めるために、次の実装を見てみましょう-

def solve(nums):
   n = len(nums)
   ans = []
   i = 0
   while i < n:
      i0 = i
      while i < n and nums[i] == nums[i0]:
         i += 1
      if (i0 == 0 or nums[i0] > nums[i0 - 1]) and (i == n or nums[i0] > nums[i]):
         if i0 != 0 or i != n:
            ans.extend(range(i0, i))
   return ans

nums = [5, 8, 8, 8, 6, 11, 11]
print(solve(nums))

入力

[5, 8, 8, 8, 6, 11, 11]

出力

[1, 2, 3, 5, 6]