[Python/파이썬]python 입력 받기

백준 문제를 풀다보면 여러 개를 한 번에 입력받아야 하는 경우가 많은데,

나는 뒤돌아서면 또 까먹기 때문에 이번 기회에 정리해 두려고 한다..

---

 

한 개 입력받기

import sys

# 개행문자 포함
a = sys.stdin.readline()

#개행문자 미포함
b = input()

print(a)
print(b)

 

a, b 순의 출력 결과

변수 a 와 b 에 dobby is free 라는 문장을 입력 받고 a, b 의 순서로 출력해보면 sys.stdin.readline 에 개행문자가 포함되어 한 줄 띄워져서 출력되는 것을 볼 수 있다.

 

import sys

# 개행문자 포함
a = sys.stdin.readline()

#개행문자 미포함
b = input()

print(b)
print(a)

 

b, a 순의 출력 결과

차이를 확실하게 알기 위해 이번에는 b, a 의 순서로 출력을 해보았다. input 에는 개행문자가 포함되어 있지 않기 때문에 위와 다르게 붙어서 출력된 것을 볼 수 있다. 

 

덧붙여서 sys.stdin.readline 은 input 보다 훨씬 빠르기 때문에 알고리즘 문제 풀 때 시간 줄이기 좋다!

 

여러 개 입력 받기

- 정해진 개수 입력 받기

import sys

# 빠름
a, b = map(int, sys.stdin.readline().split())

# 느림
c, d = map(int, input().split())

 

 

- 입력 값(문자열x) 여러 개를 리스트에 저장

import sys

# 빠름
a = list(map(int, sys.stdin.readline().split()))

# 느림
b = list(map(int, input().split())

 

- 문자열 여러 개 입력 받아 리스트에 저장

import sys

# 문자열 3개 입력받기

# 빠름
a = [sys.stdin.readline().rstrip().split() for _ in range(3)]

# 느림
b = [input() for _ in range(3)]