[Python] 백준 6322번 직각 삼각형의 두 변

문제

컴퓨터를 이용하면 수학 계산이 조금 쉬워진다. 

다음과 같은 예를 살펴보자. 세 변의 길이가 a, b, c(c는 빗변)이면서 a2+b2=c2를 만족하는 삼각형을 직각삼각형이라고 한다. 

이 공식은 피타고라스의 법칙이라고 한다.

직각 삼각형의 두 변의 길이가 주어졌을 때, 한 변의 길이를 구하는 프로그램을 작성하시오.

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

6322번: 직각 삼각형의 두 변

입력

입력은 여러 개의 테스트 케이스로 이루어져 있다. 

각 테스트 케이스는 한 줄로 이루어져 있고, 직각 삼각형의 세 변의 길이 a, b, c가 주어진다. 

a, b, c중 하나는 -1이며, -1은 알 수 없는 변의 길이이다. 

다른 두 수는 10,000보다 작거나 같은 자연수이다.
입력의 마지막 줄에는 0이 세 개 주어진다. 

출력

각 테스트 케이스에 대해서, 입력으로 주어진 길이로 직각 삼각형을 만들 수 있다면, "s = l"을 출력한다.

s는 길이가 주어지지 않은 변의 이름이고, l은 길이이다.

l은 소수점 셋째 자리까지 출력한다. 삼각형을 만들 수 없는 경우에는 "Impossible."을 출력한다.

예제

# input
3 4 -1
-1 2 7
5 -1 3
0 0 0
# output
Triangle #1
c = 5.000

Triangle #2
a = 6.708

Triangle #3
Impossible.

문제 풀이

cnt = 1

while True:
    a, b, c = map(int, input().split())
    if a == 0 and b == 0 and c == 0: break
    if cnt != 1: print()
    
    print(f'Triangle #{cnt}')
    
    if c == -1:
        print(f'c = {(a**2+b**2)**0.5:.3f}')
    elif max(a, b) >= c:
        print('Impossible.')
    elif a == -1:
        print(f'a = {(c**2-b**2)**0.5:.3f}')
    elif b == -1:
        print(f'b = {(c**2-a**2)**0.5:.3f}')
        
    cnt += 1

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.