Python에서 input()과 sys.stdin.readline()의 차이

2025.04.23 - [coding test/백준] - [Python] 2667번 단지번호붙이기

[Python] 2667번 단지번호붙이기

해당 문제를 풀면서 input()과 sys.stdin.readline()의 차이로 에러가 발생했었다.

 

에러가 발생했던 코드:

import sys
input = sys.stdin.readline
graph.append(list(map(int, input())))

이렇게 하면 input()은 줄 끝에 \n(엔터 문자)도 포함한 문자열을 받아오게 된다.

ex) 왜냐면 마지막 입력값을 치게 되면서 엔터를 누르게 되므로 입력값이 1234라면 input()값은 '1234'+'\n'가 된다.

input() == '1234\n'
list(map(int, input())) → [1, 2, 3, 4, ???] ❌

→ 여기서 \n도 문자 하나로 포함되니까
int('\n')을 시도하게 되고, ValueError가 발생 ❗

 

이를 해결하기 위해 하기 위해서는 공백 문자를 제거해주는 strip()를 쓴다.

✅ 해결 방법: .strip() 사용

input() == '1234\n'
input().strip() == '1234'
map(int, input().strip()) → [1, 2, 3, 4] ✅