[Python] 백준알고리즘 15552

빠른 A+B
본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다. 입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.
Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다. 단, 이때는 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.

첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.

각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.

5
1 1
12 34
5 500
40 60
1000 1000

2
46
505
100
2000

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

위 코드는 jupyter notebook에서 실행시 오류가 발생한다.
jupyter notebook에는 stdin 이 제대로 구성되어 있지 않기 때문에 stdin.readline() 을 실행하면 입력을 받지 못하고 항상 빈 문자열이 반환된다.

따라서, sydin.readline() 대신에 input() 을 사용해야 한다.

단계별로 풀어보기-for문