백준 15552번 빠른 A+B - map()과 readline()

백준 15552번 / 빠른 A+B : https://www.acmicpc.net/problem/15552

 

이 문제는 백준 10950번 문제와 유사하게 여러 테스트 케이스에 대해 두 정수 A와 B의 합을 구하는 것이지만, "빠른"이라는 수식어가 붙은 만큼 입력 방식에 주의해야 하는 문제이다. 

 

Python에서 많은 양의 입력을 받을 때 input() 함수는 sys.stdin.readline()에 비해 느릴 수 있어 시간 초과가 발생할 수 있다.

 

따라서 이번에는 readline()를 이용하여 a+b를 수행하는 코드를 만들어볼 것이다.

 

두 가지 코드를 만들어보았다.

 

첫 번째는 readline()만 사용한 코드이다.

import sys
N = int(input())

for i in range(N):
    a,b = sys.stdin.readline().rstrip().split()
    a_1 = int(a)
    b_1 = int(b)
    print(a_1 + b_1)

 

 

두 번째는 map()과 readline()을 같이 사용한 코드이다.

import sys
N = int(input())

for i in range(N):
     a, b = map(int, sys.stdin.readline().rstrip().split())
     print(a + b)

 

두 번째 코드가 보다 간소하기에, 이와 같은 방식을 사용하는 것이 좋아보인다.