Написати програму для відображення фігур (див. рисунок)

import math

p = int(

b = false

for i in range(1, p ** 2):

    gip = (p ** 2) + (i ** 2)

    if math.sqrt(gip) % 1 == 0:

        b = true

        if int(math.sqrt(gip)) ! = p:

            print(int(math.sqrt(

if b == false:

    print(-1)

# что-то типо этого, но не проходит по времени. надо смотреть диапазон. может если кто дорешает, ответьте. а так всё норм работает)

Следует отметить, что 1 условие выполняется во всех цепочках, а значит его можно не рассматривать.
Рассмотрим каждую комбинацию отдельно:
HDEE
2 условие - true
3 условие - true
4 условие - true
Подходит
HHAE
2 условие - true
3 условие - false(H на 2 месте)
HEAE
2 условие - true
3 условие - true
4 условие - true
Подходит
AHAH
2 условие - true
3 условие - true
4 условие - true
Подходит
AEAD
2 условие - true
3 условие - true
4 условие - true
Подходит
AEED
2 условие - false(E на 3 месте)
CAEH
2 условие - false(A не входит в список допустимых значений)
EHAD
2 условие - true
3 условие - false(E не входит в список допустимых значений)
CDEA
2 условие - true
3 условие - true
4 условие - false(A не входит в список допустимых значений)
ответ: 4