[Python] 백준 15552번 | sys.stdin.readline() & rstrip()

백준 15552 빠른 A+B

sys.stdin.readline()

반복문으로 여러 줄을 입력 받아야 할 때는 input()으로 입력 받을 때 시간 초과가 발생할 수 있다.
그 대신 사용하는 것이 sys.stdin.readline()
이때 stdin은 standard input을 뜻한다.

한 개의 정수를 입력 받을 때

import sys
a = int(sys.stdin.readline())

| Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다. 단, 이때는 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.
백준 문제 설명에도 나와 있듯이 sys.stdin.readline()를 사용하면 개행문자 \n까지 포함되어 출력되므로, int() 변환을 해주어야 한다.

정해진 여러 개의 정수를 입력 받을 때

import sys
a, b, c = map(int, sys.stdin.readline().split())

정해진 여러 개의 정수를 입력 받을 때

import sys
data = list(map(int, sys.stdin.readline().split()))

추가로, rstrip()

rstrip([chars])'는 인자로 전달된 문자를 String의 오른쪽에서 제거해준다. 추가로strip([chars])는 왼쪽과 오른쪽에서,lstrip([chars])'는 왼쪽에서 제거해준다.
sys.stdin.readline()을 사용하면 \n이 오른쪽에 같이 나와서 제거해주라고 rstrip() 쓰라고 권유해준 듯.

공백 제거

text = ' 데이터 요리조리 '
print('[' + text.rstrip() + ']')

결과

[ 데이터 요리조리]

동일한 문자 제거

text = ' 11데이터 요리조리1111 '
print('[' + text.rstrip() + ']')

결과

[ 11데이터 요리조리]

여러 문자 제거

text = ",,,,,123.....data....pp"
print(text.rstrip(',123.p'))

결과

[,,,,,123.....data]

위의 두 함수를 활용해서 [백준 15552번 - 빠른 A+B]를 풀어보자!

빠른 A+B

시간 제한 메모리 제한 제출 정답 맞힌 사람 정답 비율
1 초 (하단 참고) 512 MB 240486 108471 88503 45.732%

문제

본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다. 입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.

입력

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

출력

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

내 풀이

import sys

N = int(input())
a = [sys.stdin.readline().strip('\n') for _ in range(N)]

for i in range(N) :
    print(sum(map(int, a[i].split())))

간단 정답

import sys

N = int(input())

for i in range(N) :
    print(sum(map(int, sys.stdin.readline().split())))