def solve(N, A):
    prefix_freq = [{} for _ in range(N)]
    freq = {}

    for q in range(N):
        prefix_freq[q] = freq.copy()
        freq[A[q]] = freq.get(A[q], 0) + 1

    count = 0
    for r in range(2, N - 1):
        for s in range(r + 1, N):
            if A[r] == 0:
                if A[s] == 0:
                    count += (r * (r - 1)) // 2
                continue
            
            for q in range(1, r):
                if A[q] == 0 and A[s] == 0:
                    count += prefix_freq[q].get(0, 0)
                    continue
                
                qs = A[q] * A[s]
                if qs % A[r] != 0:
                    continue
                
                target = qs // A[r]
                count += prefix_freq[q].get(target, 0)

    return count

💻🔘 MCQ + Coding Help 🔘 💻