[파이썬] RecursionError가 뜰때, 입력을 받을때 - sys

주로 알고리즘 문제 풀 때, 안 쓰면 섭섭한 sys라이브러리

자주 쓰는 기능들을 몇 가지 적어둠.

---

 

sys.setrecursionlimit()

파이썬에서 재귀를 사용해서 문제를 풀 때 무조건 필요한 함수.

파이썬은 기본적으로 재귀 제한이 매우 적은 편이기에 문제를 풀 때 자주 런타임 에러가 발생한다.

재귀가 막힌다 싶으면 무조건 작성해야 함.

import sys
sys.setrecursionlimit(10**6)

 

---

sys.stdin.readline()

주로 입력을 빨리 받고싶을때 input() 대신 사용한다.

반복문으로 입력을 많이 받을때 sys.stdin.readline()을 사용해주면 시간을 단축시킬 수 있다.

 

import sys

arr = []
arr.append(sys.stdin.readline())
print(arr)
# 입력 abc
# 출력 ['abc\n']

sys.stdin.readline()은 한 줄 단위로 받기 때문에 개행 문자 \n도 같이 입력받아진다.

그래서 rstrip()을 사용해서 개행문자를 지워준다.

그리고 문자열형태로 받기 때문에 정수로 사용하려면 형 변환도 필요하다.

 

import sys

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

정수를 한줄로 받은 경우

 

import sys

arr = list(map(int,sys.stdin.readline().rstrip().split()))

정수를 한줄로 받아 리스트로 만들 때

 

import sys
arr2 = [int(x) for x in sys.stdin.readline().rstrip().split()]

정수를 한 줄로 받아 리스트로 만들 때 - 리스트 컴프리헨션 사용

 

import sys
arr3 = [int(sys.stdin.readline().rstrip()) for i in range(3)]

정수를 여러 줄 받을 때

입력

1

2

3

arr3 = [1, 2, 3]

 

import sys
arr3 = [list(map(int,sys.stdin.readline().rstrip().split())) for i in range(3)]
print(arr3)

정수를 2차원 배열로 받을 때(복잡해 보이므로 for문을 쓰는 게 좋을 거 같기도 하다)

입력

1 3 5
2 4 6
7 8 9

arr3 = [[1, 3, 5], [2, 4, 6], [7, 8, 9]]