ALGORITHM/PYTHON

백준 BAEKJOON 31798번 단원평가  [PYTHON/파이썬]

칼코
반응형

 

 

 

 

 

 

백준 BAEKJOON 31798번 단원평가 [PYTHON/파이썬]


<문제 출처> (BRONZE Ⅲ)

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

 

 

 

 

 

 

<풀이>

위의 수 두 개 중 하나라도 0이라면

아래 수의 제곱에서 나머지 수를 빼주면 된다.

아래 수가 0이라면

위의 수 두 개를 더한 뒤 루트를 씌워주면 된다.

루트는 1/2 제곱을 해주면 된다.

 

 

 

 

 

 

 

<코드>

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

if a == 0:
    print(int(c**2 - b))
elif b == 0:
    print(int(c**2 - a))
elif c == 0:
    print(int((a + b) ** (1 / 2)))

 

 

 

 

 

 

 

 

반응형