코딩테스트

백준 입출력 문제 - try except문 활용(Python)

수연 (Suyeon) 2023. 11. 25. 00:19
반응형

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

 

10951번: A+B - 4

두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

www.acmicpc.net

 

10951번 문제처럼 횟수가 정해지지 않고 입출력을 하려고 했을 때 while을 활용했지만 아래처럼 해서 오류가 발생했습니다.

while True:
  A, B = map(int, input().split())
  if(!A && !B) break;
  print(A + B)

이건 input()을 했을 때 값이 입력되지 않으면 NameError가 발생하는 것을 모르고 코드를 구현했었습니다.

그 후, 오류가 발생할 경우에는 try-except문으로 해결하면 된다는 것을 배웠고 아래처럼 코드를 수정했습니다.

 

while True:
  try:
    A, B = map(int, input().split())
    print(A + B)
  except:
    break

try에서는 오류가 발생하지 않을 때 안에 있는 코드들이 동작하고, 오류가 발생하면 except문 안에 있는 코드가 작동됩니다.

728x90