Chương trình tìm số cặp tốt trong Python

Cho mảng nums. Cặp (i,j) gọi là tốt nếu nums[i] == nums[j] và i < j. Đếm số cặp tốt.

Ví dụ: nums = [5,6,7,5,5,7] → 4 cặp: (0,3), (0,4), (3,4), (2,5).

Thuật toán

• Khởi tạo count = 0.
• Hai vòng lặp lồng nhau i từ 0 đến n-1, j từ i+1 đến n-1.
• Nếu nums[i] == nums[j], tăng count.
• Trả về count.

Ví dụ (Python)

def solve(nums):
    count = 0
    n = len(nums)
    for i in range(n):
        for j in range(i+1, n):
            if nums[i] == nums[j]:
                count += 1
    return count

nums = [5,6,7,5,5,7]
print(solve(nums))

Đầu vào

[5,6,7,5,5,7]

Đầu ra

4