주피터 노트북에서 백준 입력 편하게 받기

백준에서 입력값이 여러 줄일 경우 일일이 입력하는게 상당히 귀찮다.

이럴때 로컬 경로의 텍스트 파일에서 직접 입력을 받으면 훨씬 편하다.

해당 주피터 노트북 파일(.ipynb)이 실행되는 경로와 같은 위치에 텍스트 파일을 생성한 후,

코드의 앞 부분에 한 줄을 추가하면 직접 입력해 input을 받는 코드와 완전히 동일하다.

(기존의 input 함수를 덮어쓰는 것에 유의)

# input.txt에 입력된 두 정수의 합을 출력
input = open("input.txt").readline

a, b = map(int, input().split())
print(a + b)

# input 함수 원래대로
# del input

input.txt는 다른 이름으로 대체가 가능하고, input 역시 기존의 input 함수를 덮어쓰는게 싫으면 다른 이름으로 지정해도 된다.

input함수를 원래대로 되돌리고 싶으면 del input을 실행한다.

파이썬 파일(.py)에서 작업한다면 이 방법 외에도 sys.stdin으로 가능하지만 주피터 노트북에서는 불가능하다.