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

Chương trình để kiểm tra chúng ta có thể tìm thấy ba phần tử duy nhất có tổng giống như k hay không trong Python

Giả sử chúng ta có một danh sách các số được gọi là num và một giá trị khác k, chúng ta phải kiểm tra xem chúng ta có thể tìm thấy ba phần tử duy nhất trong danh sách có tổng là k hay không.

Vì vậy, nếu đầu vào giống như nums =[11, 4, 6, 10, 5, 1] ​​k =20, thì đầu ra sẽ là True, vì chúng ta có các số [4,6,10] có tổng là 20.

Để 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

  • l:=0, r:=kích thước của nums - 1

  • trong khi l

    • t:=k - nums [l] - nums [r]

    • nếu nums [r - 1]

      • l:=l + 1

      • đi ra từ vòng lặp

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

      • nếu nums [m]> t, thì

        • r:=r - 1

        • đi ra từ vòng lặp

      • nếu nums [m] giống với t thì

        • trả về True

  • trả về Sai

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, k):
      nums.sort()
      l, r = 0, len(nums) − 1
      while l < r − 1:
         t = k − nums[l] − nums[r]
         if nums[r − 1] < t:
            l += 1
            continue
         for m in range(l + 1, r):
            if nums[m] > t:
               r −= 1
               break
            if nums[m] == t:
               return True
      return False
ob1 = Solution()
nums = [11, 4, 6, 10, 5, 1]
k = 20
print(ob1.solve(nums, k))

Đầu vào

[11, 4, 6, 10, 5, 1], 20

Đầu ra

True