どん底から這い上がるまでの記録

どん底から這い上がりたいけど這い上がれない人がいろいろ書くブログ(主にプログラミング)

Is it a Right Triangle?

 

直角三角形

問題ページ

解き方

3辺の長さを受け取った後ソートする。
一番長い辺をc、それ以外をそれぞれa, bとしa²+b²=c²が成り立てば直角三角形となる。

コード(python)

def triangle(dataset):
    a, b, c = dataset
    if a*a + b*b == c*c:
        return 1
    return 0

N = int(input())
for _ in range(N):
    dataset = sorted(map(int, input().split()))
    flag = triangle(dataset)
    print(["NO", "YES"][flag])