[Python]피타고라스의 정리

<문제 설명>
직각삼각형이 주어졌을 때 빗변의 제곱은 다른 두 변을 각각 제곱한 것의 합과 같습니다.
직각삼각형의 한 변의 길이를 나타내는 정수 a와 빗변의 길이를 나타내는 정수 c가 주어질 때, 다른 한 변의 길이의 제곱, b_square 을 출력하도록 한 줄을 수정해 코드를 완성해 주세요.

입출력 예 설명
입출력 예 #1 a2 = 9, c2 = 25 이므로 16을 출력합니다.
입출력 예 #2 a2 = 81, c2 = 100 이므로 19를 출력합니다.

 

a = int(input())
c = int(input())

b_square = c**2 - a**2
print(b_square)