Computer >> Máy Tính >  >> Lập trình >> Python

Chương trình kiểm tra số bộ ba từ một mảng có tổng nhỏ hơn đích hoặc không phải Python

Giả sử chúng ta có một danh sách các số được gọi là nums và một mục tiêu giá trị khác, chúng ta phải tìm số bộ ba (i

Vì vậy, nếu đầu vào giống như nums =[−2, 6, 4, 3, 8], target =12, thì đầu ra sẽ là 5, vì bộ ba là:[−2,6,4], [−2 , 6,3], [−2,4,3], [−2,4,8], [−2,3,8]

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

  • sắp xếp số lượng danh sách

  • ans:=0

  • n:=kích thước của nums

  • đối với tôi trong phạm vi từ 0 đến n-1, hãy thực hiện

    • k:=n - 1

    • đối với j trong phạm vi i + 1 đến n − 1, thực hiện

      • while k> j và nums [i] + nums [k] + nums [j]> =target, do

        • k:=k - 1

      • nếu j giống k thì

    • ans:=ans + k - j

  • 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, target):
      nums.sort()
      ans = 0
      n = len(nums)
      for i in range(n):
         k = n − 1
         for j in range(i + 1, n):
            while k > j and nums[i] + nums[k] + nums[j] >=
target:
               k -= 1
            if j == k:
               break
            ans += k − j
      return ans
ob1 = Solution()
nums = [−2, 6, 4, 3, 8]
target = 12
print(ob1.solve(nums, target))

Đầu vào

[-2, 6, 4, 3, 8], 12

Đầu ra

5