Kiểm tra hiệu diện tích hai hình vuông có phải số nguyên tố trong Python

Cho hai số x và y. Kiểm tra xem hiệu diện tích hai hình vuông cạnh x và y (tức x^2 - y^2) có phải số nguyên tố hay không.

Quan sát: x^2 - y^2 = (x-y)(x+y). Để tích này là số nguyên tố, một thừa số phải bằng 1. Do x+y > x-y, nên x-y = 1 và x+y phải là số nguyên tố.

Ví dụ: x=7, y=6 → x-y=1, x+y=13 (nguyên tố) → True.

Giải thuật

• Kiểm tra x - y == 1 và x + y là số nguyên tố.

Mã ví dụ

def is_prime(num):
    if num <= 1:
        return False
    if num <= 3:
        return True
    if num % 2 == 0 or num % 3 == 0:
        return False
    i = 5
    while i * i <= num:
        if num % i == 0 or num % (i + 2) == 0:
            return False
        i += 6
    return True

def solve(x, y):
    if is_prime(x + y) and x - y == 1:
        return True
    else:
        return False

x, y = 7, 6
print(solve(x, y))

Đầu vào

7,6

Đầu ra

True