【每日一练】—— 勾股数

[爱吃饼干的小白鼠]

勾股数

 

勾股数是一组三个正整数，它们可以作为直角三角形的三条边。
比如3 4 5就是一组勾股数。
如果给出一组勾股数其中的两个，你能找出余下的一个吗？

输入描述：

第一行输入两个整数n。(1<=n<=1000000)表示一组勾股数其中的两个。

输出描述：

如果余下的一个勾股数存在，则输出之。
如果不存在，则输出-1。

输入样例：

5 6

输出样例：

-1

 

from math import sqrt

a, b = map(int, input().split())

if a == b:
    print(-1)


if sqrt(a * a + b * b) == int(sqrt(a * a + b * b)):
    print(int(sqrt(a * a + b * b)))


if sqrt(abs(a * a - b * b)) == int(sqrt(abs(a * a - b * b))):
    print(int(sqrt(abs(a * a - b * b))))


else:
    print(-1)

 

[CSDN-Ada助手]

您的问题已经帮您同步至问答, 链接: https://ask.csdn.net/questions/7953076, 请您保持关注, 如果回答有帮助解决此问题, 麻烦您动动小手给相关的回答点赞, Ada会在评论区为您更新结题状态