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

Tìm ba phần tử từ ba mảng khác nhau sao cho a + b + c =sum trong Python


Giả sử chúng ta có ba mảng A, B, C và một giá trị khác gọi là "sum", chúng ta phải kiểm tra xem có ba phần tử a, b, c sao cho a + b + c =sum và a, b và c phải nằm dưới ba mảng khác nhau.

Vì vậy, nếu đầu vào giống như A =[2,3,4,5,6], B =[3,4,7,2,3], C =[4,3,5,6,7], tổng =12, thì đầu ra sẽ là True vì 4 + 2 + 6 =12 và 4, 2, 6 lần lượt được lấy từ A, B, C.

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

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

    • đối với j trong phạm vi từ 0 đến kích thước của B, thực hiện

      • đối với k trong phạm vi từ 0 đến kích thước của C, thực hiện

        • nếu A [i] + B [j] + C [k] giống với tổng thì

          • trả về True

  • trả về Sai

Ví dụ

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

def is_sum_from_three_arr(A, B, C, sum):
   for i in range(0 , len(A)):
      for j in range(0 , len(B)):
         for k in range(0 , len(C)):
            if (A[i] + B[j] + C[k] == sum):
               return True
   return False
A = [2,3,4,5,6]
B = [3,4,7,2,3]
C = [4,3,5,6,7]
sum = 12
print(is_sum_from_three_arr(A, B, C, sum))

Đầu vào

[2,3,4,5,6], [3,4,7,2,3], [4,3,5,6,7], 12

Đầu ra

True