[알고리즘] 입력을 빠르게 받는 방법
입력을 빠르게 받기
문자열
한 줄
Python
import sys
s = sys.stdin.readline().rstrip()
- 주의사항
sys.stdin.readline()은 끝에 개행문자(\n)가 포함돼서.rstrip()또는.strip()으로 지워야 해요..rstrip()은 오른쪽 끝의 공백과 개행을,.strip()은 양쪽 끝을 지워요. 개행만 지우고 싶다면.rstrip('\n')을 쓰면 돼요.
리스트
Python
import sys
char_list= list(sys.stdin.readline().rstrip())
- 위처럼 하면 한 글자(char)씩 담긴 리스트를 얻을 수 있어요.
Python
import sys
word_list = sys.stdin.readline().rstrip().split()
- 입력이 공백으로 나뉘어 있다면, 위처럼 한 단어씩 담긴 리스트를 얻을 수 있어요.
숫자
한 줄
Python
import sys
a,b = map(int, sys.stdin.readline().split())
- 참고 사항
split()은 인자로 받은 구분자로 문자열을 나누는 함수예요. 인자를 비워 두면 공백(스페이스, 탭, 개행)을 기준으로 나누고, 연속된 공백은 하나로 취급해요.