Python sys.stdin.readline() 사용

Question

안녕하세요. 구간합구하기 (백준 11660) 문제를 풀고 있는데요. 코드는 똑같은데, 제일 상단에 아래 코드 입력 여부에 따라 시간초과/성공이 갈립니다. import sys input = sys.stdin.readline 이게 실패와 성공을 가를만큼 중요한 역할을 하나요? 위 코드를 넣으면 왜 빨라지는지 설명해주실 수 있으실까요?

communityai8509 · Answer

안녕하세요, 인프런 AI 인턴입니다. 네, input = sys.stdin.readline 코드는 Python에서 입력을 받는 속도를 향상시켜주는 역할을 합니다. 기본적으로 Python의 input() 함수는 사용자의 입력을 받으면서, 공백문자(예: 줄바꿈)를 제거하는 등의 추가적인 작업을 수행하기 때문에 상대적으로 느릴 수 있습니다. 반면 sys.stdin.readline 은 줄바꿈 문자를 포함한 한 줄의 문자열을 빠르게 입력받기 때문에, 많은 양의 데이터를 빠르게 입력받아야 하는 경우 (예: 알고리즘 문제에서의 대량의 데이터 입력)에 훨씬 효율적입니다. 다만, sys.stdin.readline 을 사용할 때에는 문자열의 마지막에 줄바꿈 문자가 포함되기 때문에, 이를 제거하기 위해 종종 strip() 을 사용해야 할 수 있습니다. 따라서 알고리즘 문제를 풀 때 입출력의 양이 많고 빠른 입출력이 요구되는 경우, input() 대신 sys.stdin.readline 을 사용하여 시간 초과를 방지할 수 있습니다. 문제 푸시느라 수고 많으십니다. 질문이 더 있으시면 언제든 질문해주세요.