Algorithm/Beakjoon

[Python] 백준 10951번_A+B - 4

mopipi 2022. 7. 30. 21:44
반응형

문제 

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

 

10951번: A+B - 4

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

www.acmicpc.net

코드

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

 

풀이 

우선 처음 문제를 접했을 때 당황스러웠다.. 테스트 케이스에 대한 세부적인 설정이 없었기 때문...

몇 번을 반복해야 하는지 갈피를 못잡고, 우선 while True: 로 무한 반복을 돌렸는데, 당연히 오류가 났다ㅎ

 

테스트 케이스 횟수에 대한 특정 언급이 없는 경우에는, 테스트 케이스가 입력되지 않는 경우를 끝으로 가정한다.

try - except 문을 이용해 케이스가 입력되지 않는 경우 발생하는 예외를 계산의 끝으로 처리할 수 있다.

 

+

출처: https://www.acmicpc.net/source/16590776

# A+B-6 문제
import sys

for i in range(int(sys.stdin.readline())):
    num = sys.stdin.readline().split(",")
    print(int(num[0])+int(num[1]))

sys를 import해 input 대신 readline을 하면 실행 속도를 절약할 수 있다.

또한 각각의 변수 a, b에 할당해 저장하는 대신 list를 이용해 저장하고 인덱스로 접근 가능하다. 

(리스트에는 char형태로 저장한 후, 나중에 계산할 때 int형으로 바꿔줬다)

반응형