[python] 백준 문제풀이를 위한 입출력 방법 (sys.stdin.readline(), input() 차이점)

1️⃣ 입력값 한 개

- input()은 기본적으로 문자열로 받는다.

a = input()
# 입력: hello, 출력: hello (string)

b = int(intput())
# 입력: 1, 출력: 1 (int형)

2️⃣ 입력값을 각각 변수에 넣어주기 

# 정수
a, b = map(int, input().split())

# 문자열
a, b = input().split()

3️⃣  입력값을 1차원 배열로

- 한 번에 입력받고 잘라서 1차원 배열에 넣어주는 방법
- split(): 공백을 기준으로 문자열을 잘라 배열에 넣음. 괄호 안에 자르는 기준을 넣어줄 수 있음.

# ✅ string인 경우
a = input().split()     #hello my name is gamza
b = input().split(':')  #hello:my name is: gamza

print(a)    #['hello', 'my', 'name', 'is', 'gamza']
print(b)    #['hello', 'my name is', 'gamza']

####################################################
# ✅ int형 으로 입력을 받아야하는 경우
a = list(map(int, input().split()))   # 띄어쓰기로 입력

print(a)  #[1,2,3,4,5]

- 여러번 입력받아서, 입력받은 값으로 1차원 배열을 만들어주는 방법

# 1️⃣ 첫번째 방법: for문 이용
n = int(input())      # 5
arr = []
for i in range(0, n):
    arr.append(int(input()))   # 엔터기준으로 5번 입력받음 (1, 2, 3, 4, 5)

print(arr)    #[1, 2, 3, 4, 5]


# 2️⃣ 두번째 방법: 바로 받아서 배열로 넣어주기
n = int(input())    # 5
arr = [int(input()) for _ in range(n)]   # 엔터기준으로 5번 입력받음 (1, 2, 3, 4, 5)
print(arr)    #[1, 2, 3, 4, 5]

4️⃣ 여러 개를 2차원 배열로 입력받기

- [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 와 같은 형태로 입력받기

arr = [list(map(int,input().split())) for _ in range(n)]

- n: arr의 length
- int 빼면 string으로 가능
- split() 빼면 한문자씩 나눠줌 

5️⃣ sys 사용하기

- 위에서 작성된 모든 input이 들어가는 자리에 sys.stdin.readline() 을 입력하면 된다. 

import sys
num = int(sys.stdin.readline())
arr = [list(sys.stdin.readline().split()) for _ in range(num)]

print(arr)

## num: 3
## arr : [['1', '2', '3'], ['4', '5', '6'], ['7', '8', '9']]

 

🤔 sys.stdin.readline() 과 input() 의 차이점 ?_?

1. 개행문자

input()는 자체적으로 개행문자(엔터, \n)를 삭제시켜 반환하고, sys.stdin.readline()은 포함하여 반환한다.

import sys

str1 = input()
str2 = sys.stdin.readline()

print(str1)
print(str2)
print("------end--------")

sys.stdin.readline()으로 입력된 str2다음에 엔터가 포함되고 end가 나오는 것을 볼 수 있다.

2. prompt message

input()은 파라미터로 prompt message를 받는다. 그래서 실행될 때 사용자 입력을 받아들이기 전, prompt message를 먼저 실행하고 입력을 받는다.

str1 = input("입력해주세요 \n")
print(str1)
print("------end--------")

3. 버퍼에 저장하는 방식이 다르다. 

sys.stdin.readline은 사용자 입력 값을 한번에 받아와 버퍼에 모두 저장한다. 반면에 input()은 값을 입력할 때마다 따로따로 버퍼에 저장한다. 따라서 버퍼 사이즈의 차이가 발생하므로, 입력이 반복될 수록 input보다 sys의 속도는 확연하게 빨라진다.

 

✅ 따라서 위의 이유들 때문에 input()은 sys.stdin.readline()보다 시간이 오래걸릴 수 밖에 없다. 따라서, 백준과 같은 알고리즘 문제를 풀떄는 sys모듈을 사용한 readline()을 사용하는게 더 합리적이다.