[백준/Python] 15552번 빠른 A+B 문제

■ 15552번 빠른 A+B 문제

출처: 백준 15552번 빠른 A+B

 

■ 코드 풀이

 

아이디어는 다음과 같습니다.

1) 파이썬에서 데이터 입력은 대부분 input()으로 알고 있지만, sys.stdin도 input()과 동일한 기능을 합니다. stdin은 standard input의 약자로, input() 대비 빠르고 메모리 효율적입니다. 따라서, 코딩 테스트에서는 input() 대신 stdin을 활용하여 소요되는 시간과 메모리를 절약할 수 있습니다. (실제로 대다수의 문제에서 input()을 사용하면 오답이지만, stdin()을 사용하면 정답인 경우가 많습니다.)

2) stdin을 활용하기 위해 sys 모듈을 import 해줍니다. 이후, input과 동일하게 T에 반복 횟수를 입력 받습니다.

3) 반복 횟수 T 동안 A,B에 두 정수를 입력 받습니다. map은 첫 번째 인자로 적용할 함수를, 두 번째 인자로 list 같은 iterable을 입력 받습니다. sys.stdin.readline().split()은 입력 받은 정수 값을 공백 문자로 split하여 list로 반환하므로 map에 활용할 수 있습니다.

4) 더한 값을 출력합니다.

import sys

T = int(sys.stdin.readline())

for i in range(T):
    A,B = map(int, sys.stdin.readline().split())
    print(A+B)

 

참고로 아래는 sys.stdin.readline() 대신 input()을 사용한 경우입니다. 햇갈리시면 위의 코드와 비교하며 공부하셔도 좋을 듯합니다.

T = int(input())

for i in range(T):
    A,B = map(int, input().split())
    print(A+B)