Kiểm tra mảng có phần tử bằng tích các phần tử còn lại trong Python

Cho mảng nums. Kiểm tra xem có tồn tại phần tử nào mà giá trị của nó bằng tích của tất cả các phần tử còn lại hay không.

Ví dụ: [3,2,24,4,1] → 24 = 3*2*4*1 → True.

Giải thuật

• Tính tích tất cả phần tử mul.
• Duyệt mảng, nếu nums[i] == mul / nums[i] → True.

Mã ví dụ

def solve(nums):
    mul = 1
    for i in range(len(nums)):
        mul *= nums[i]
    for i in range(len(nums)):
        if nums[i] == mul / nums[i]:
            return True
    return False

nums = [3,2,24,4,1]
print(solve(nums))

Đầu vào

[3,2,24,4,1]

Đầu ra

True