[알고리즘] 입력을 빠르게 받는 방법

입력을 빠르게 받기

문자열

한 줄

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()은 인자로 받은 구분자로 문자열을 나누는 함수예요. 인자를 비워 두면 공백(스페이스, 탭, 개행)을 기준으로 나누고, 연속된 공백은 하나로 취급해요.