[Python] 알고리즘 문제 풀이 시 필요한 주요 팁 (sys.setrecursionlimit(), sys.stdin.readline())

문제를 많이 풀어보면서 몰랐던,,! 쉽지만 중요한 팁들을 간단히 정리해보고 있다.

 

📍 재귀 깊이 제한 - sys.setrecursionlimit()

파이썬은 기본으로 재귀 깊이를 1000으로 제한한다.

따라서 재귀를 더 깊게 돌리기 위해서는 이 제한을 풀어줘야 한다.

 

푸는 방법은 간단하다.

sys 모듈을 import하고, setrecursionlimit 함수를 사용하면 된다.

import sys
sys.setrecursionlimit(10000)

알고리즘 문제를 풀 때 이유 없이 런타임 에러가 난다면 이것을 생각해봐야 할 듯 하다.

---

📍 입력 - input() / sys.stdin.readline()

입력 받을 내용이 별로 없다면 input 함수로도 충분하지만,

여러 줄의 입력을 많이 받아야 한다면 sys.stdin.readline을 사용해주는 것이 좋다.

sys.stdin.readline이 input보다 시간과 메모리 측면 모두에서 효율이 좋기 때문이다.

 

sys를 import하고 사용해주면 된다.

아래에 sys.stdin.readline의 사용 방법을 정리했다.

import sys

a = sys.stdin.readline()
print(a) # input: 1 / output: "1\n"

b = sys.stdin.readline().strip()
print(b) # input: 1 / output: "1"

c = int(sys.stdin.readline())
print(c) # input: 1 / output: 1

d, e = map(int, sys.stdin.readline().split())
print(d, e) # input: 1 2 / output: 1 2

f = [sys.stdin.readline().strip().split() for _ in range(2)]
print(f) # input: 1 2\n2 3 / output: [['1', '2'], ['2', '3']]

# (아래와 같은 표현) g = [list(map(lambda x: int(x), sys.stdin.readline().strip.split())) for _ in range(2)]
g = [list(map(int, sys.stdin.readline().split())) for _ in range(2)]
print(g) # input: 1 2\n2 3 / output: [[1, 2], [2, 3]]

👉 [ 주의할 점 ]

• input은 한 줄을 입력 받은 후, 발생한 개행문자(\n)을 제거하여 문자열로 return 하는 반면,
  sys.stdin.readline은 개행문자를 제거하지 않고 return한다는 특징이 있다.
• int로 변환 시 개행 문자는 자동으로 사라지지만, 그렇지 않을 경우에는 신경써줘야 하는데,
  strip 함수를 readline 뒤에 붙여주면 앞 뒤의 개행 문자를 없애줄 수 있다.

위의 주의할 사항 외에는 모두 input 자리에 대신 쓰면 된다고 생각하면 쉽다.