[Tip] 백준 시간 초과 에러

로직의 이슈가 없는데도 시간 초과가 난다면, input() 대신 아래와 같이 sys 모듈을 사용하자.

python 이외의 언어는 빠른 A+B를 참고하세요.

import sys

#num = int(input()) 대신 아래의 코드를 쓰자.
num = int(sys.stdin.readline())

 

하지만 우리는 코딩할 때, input()으로 코드를 짜기 때문에 이를 일일히 sys.stdin.readline() 바꾸기가 번거롭다. 그럴 때는 sys.stdin.readline 를 input 으로 덮어쓰면 조금 더 간편해진다.

import sys

input = sys.stdin.readline
#기존의 코드를 수정할 필요가 없어진다.
num = int(input())

 

참고로, sys.stdin.readline() 은 마지막에 엔터 개행문자가 들어가므로 rstirp()을 사용하여 제거해야 한다.

import sys
input = sys.stdin.readline

text = input().strip(" ").rstrip()

#rstrip() : 문자열의 오른쪽 공백제거