Chương trình tìm số dãy số học từ một danh sách các số trong Python?

Giả sử chúng ta có một danh sách các số gọi là num, chúng ta phải tìm số các dãy số liền nhau có độ dài ≥ 3. Như chúng ta đã biết, một dãy số là một danh sách các số mà hiệu của một số và số tiếp theo là như nhau.

Vì vậy, nếu đầu vào là nums =[6, 8, 10, 12, 13, 14], thì đầu ra sẽ là 4, vì chúng ta có các chuỗi số học như:[6, 8, 10] [8, 10, 12] [6, 8, 10, 12] [12, 13, 14]

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

• count:=0, ans:=0

• đối với tôi trong phạm vi 2 đến kích thước của nums, hãy thực hiện

  • nếu nums [i] - nums [i - 1] giống nums [i - 1] - nums [i - 2], thì

    • count:=count + 1

  • nếu không,

    • ans:=ans + thương số của (count * (count + 1)) / 2

    • đếm:=0

• nếu số đếm khác 0 thì

  • ans:=ans + thương số của (count * (count + 1)) / 2

• trả lại ans

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn:

Ví dụ

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))

Đầu vào

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

Đầu ra

4