Задача 58: Сумма квадратов

Условие задачи:

Проверьте, можно ли представить число в виде суммы двух квадратов натуральных чисел.

Входные данные:

На вход программе подается натуральное число \(n <=10^9\). Проверьте, можно ли представить его в виде суммы двух квадратов натуральных чисел.

Выходные данные:

Выведите YES или NO в зависимости от ответа на вопрос задачи. В случае положительного ответа во второй строке выведите два числа, сумма квадратов которых равна \(n\). Числа следует выводить в порядке неубывания.

Решение:

import math

n = int(input())

for i in range(1, math.ceil(math.sqrt(n)) + 1):
    a = n - i ** 2
    if a <= 0:
        print("NO")
        exit(0)
    b = math.sqrt(a)
    if (b).is_integer():
        print("YES")
        print(i, int(b))
        break

У вас включен блокировщик рекламы. Пожалуйста, выключите его чтобы увидеть решения. На этом сайте не так много рекламы, а кушать что-то надо. Вот котик