Pythonの数のリストから等差数列の数を見つけるプログラム？

numsと呼ばれる数のリストがあるとすると、長さが3以上の連続する等差数列の数を見つける必要があります。等差数列は、ある数と次の数の差が同じである数のリストです。

したがって、入力がnums =[6、8、10、12、13、14]のような場合、次のような等差数列があるため、出力は4になります。[6、8、10] [8、10 12] [6、8、10、12] [12、13、14]

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

• count：=0、ans：=0

• 2からnumsのサイズの範囲のiの場合、実行します

  • nums[i]-nums[i-1]がnums[i-1]-nums[i-2]と同じである場合、

    • count：=count + 1

  • それ以外の場合

    • ans：=ans +（count *（count + 1））/ 2

      の商

    • カウント：=0

• countがゼロ以外の場合、

  • ans：=ans +（count *（count + 1））/ 2

    の商

• ansを返す

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

例

class Solution:
   def solve(self, nums):
      count = 0
      ans = 0
      for i in range(2, len(nums)):
         if nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2]:
            count += 1
         else:
            ans += (count * (count + 1)) // 2
            count = 0
      if count:
         ans += (count * (count + 1)) // 2
      return ans

ob = Solution()
nums = [6, 8, 10, 12, 13, 14]
print(ob.solve(nums))

入力

[6, 8, 10, 12, 13, 14]

出力

4