[Algorithm] 백준 4153번 직각삼각형 (Python)

https://www.acmicpc.net/problem/4153

4153번: 직각삼각형

문제

과거 이집트인들은 각 변들의 길이가 3, 4, 5인 삼각형이 직각 삼각형인것을 알아냈다. 주어진 세변의 길이로 삼각형이 직각인지 아닌지 구분하시오.

입력

입력은 여러개의 테스트케이스로 주어지며 마지막줄에는 0 0 0이 입력된다. 각 테스트케이스는 모두 30,000보다 작은 양의 정수로 주어지며, 각 입력은 변의 길이를 의미한다.

출력

각 입력에 대해 직각 삼각형이 맞다면 "right", 아니라면 "wrong"을 출력한다.

예제 입력 1

6 8 10
25 52 60
5 12 13
0 0 0

예제 출력 1

right
wrong
right

풀이

설명

• 피타고라스의 공식을 사용하면 된다.
  • a^2 + b^2 = c^2
• c는 가장 길이가 긴 변을 의미하므로 가장 큰 값은 따로 저장하고 nums에서 삭제한다.

코드

while True:
    nums = list(map(int, input().split()))

    if sum(nums) == 0:
        break

    max_num = max(nums)
    nums.remove(max_num)

    total = nums[0]**2 + nums[1]**2
    result = max_num**2

    if total == result:
        print('right')
    else:
        print('wrong')