파이썬 입출력

이 포스트는 2021.12~2022.09 기간동안 벨로그에 작성한 글을 티스토리에 옮겨 적은 것입니다.

숫자 하나 입력받는 경우

x=int(input())

공백으로 구분된 N,M,K를 입력받는다고 하면

N,M,K=list(map(int,input().split()))

x=list(map(int,input().split()))
N=x[0]
M=x[1]
K=x[2]

이렇게 쓰지 말자..🤦‍♀️

원리: input()으로 입력받은 문자열을 split()으로 나눈 리스트로 바꾸고, map을 이용하여 해당 리스트의 모든 원소를 int로 바꾼것!

import sys
n,m=list(map(int, sys.stdin.readline().split()))
array=list(map(int,sys.stdin.readline().split()))

input()은 속도가 느리기 때문에 입력의 개수가 많은 경우에는 sys 라이브러리를 사용할 수 있다. 기존 input()의 자리에 sys.stdin.readline() 이 들어갔다고 생각하면 편하다.