백준 입출력 - 파이썬

처음 백준을 접할 때 입력, 출력 관련해서 고생을 좀 했었어서 기록을 남겨본다.

 

 

1. 입력이 한 줄에 하나씩 들어옴

 

input() 함수를 통해 입력을 받을 수 있다.

단, string으로 입력값이 받아지므로 숫자로 받고 싶다면 형 변환이 필요하다.

 

N = input() # N = "3"
N = int(input()) # N = 3

 

 

입력값이 여러 줄인 경우에도 마찬가지이다.

 

N = int(input()) # N = 5

for _ in range(N):
	print(input()) # 이후의 입력을 한 줄씩 프린트

 

파이참에서 input() 함수로 입력값을 받을 때, 백준 예제 입력을 복붙해서 엔터를 치면 마지막 줄 입력값이 반영이 안 되는 경우가 자주 있다.

이런 경우에는 그냥 엔터만 한 번 더 쳐주면 정상적으로 처리된다.

 

 

 

2. 입력값이 한 줄에 여러 개 들어옴

 

백준 1000번(A+B)이 예시이다.

a, b = input().split() # a = "1", b = "2"
lst = list(input().split()) # lst = ["1", "2"]

a, b = map(int, input().split()) # a = 1, b = 2
lst = list(map(int, input().split())) # lst = [1, 2]

 

split() 함수를 사용하면 공백, 개행 문자를 기준으로 문자열을 나눠준다.

"1 2"라는 문자열의 경우에는 중간의 공백을 기준으로 "1"과 "2"로 나눠진다.

쪼개진 값을 리스트에 담고 싶다면 list()로 감싸주면 된다.

 

숫자로 입력을 받아야 하는 경우, map() 함수를 사용하면 된다. 

map(변환할 type, 문자열)을 넣어주면 된다.(형 변환이 불가능하면 오류가 난다)

마찬가지로 리스트 형식으로 담고 싶다면 바깥을 list()로 감싸주면 된다.

 

 

 

 

3. 빠른 입력

 

보통 입력받는 값의 개수가 10만 개 정도 되는데 빠른 입력을 사용하지 않는다면 시간 초과가 날 수 있다.

빠른 입력 메서드를 사용하기 위해서는 python 파일 최상단에 sys 모듈의 stdin을 불러와야 한다.

 

from sys import stdin

ipt = stdin.readline() # ipt = "3\n"
ipt = stdin.readline().rstrip() # ipt = "3"
ipt = int(stdin.readline()) # ipt = 3

 

마찬가지로 string 형태로 입력값을 받아오기 때문에 숫자로 받고 싶다면 형 변환을 시켜줘야 한다.

readline() 함수를 사용할 때 주의해야 할 점은, 개행 문자까지 입력값으로 들어온다는 뜻이다.

 

예를 들어, "happy"라는 문자열을 stdin.readline()을 통해 읽어오면 "happy\n"으로 읽어진다.

개행 문자 때문에 문제가 생기는 것을 방지하기 위해 string type을 읽어올 때는 rstrip()을 통해 우측 개행 문자를 삭제해주는 것이 안전하다.

 

 

 

4. 출력

 

나는 보통 그냥 print() 함수를 이용해 출력한다. print 함수를 사용하면 기본적으로 줄 바꿈이 된다.

A = 3
print(A) # 3

 

줄 바꿈 없이 출력하고 싶다면(리스트를 옆으로 나란히 출력한다거나), end라는 인수를 사용하면 된다.

이 때는 출력할 값 뒤에 개행 문자("\n") 대신 end 값으로 넣어준 문자열이 추가되어 출력된다.

 

lst = [1, 2, 3]

# end 없이 출력
for i in range(len(lst)):
	print(lst[i])
# 결과
# 1
# 2
# 3

# end = " "
for i in range(len(lst)):
	print(lst[i], end=" ")
# 결과
# 1 2 3

# end = ""
for i in range(len(lst)):
	print(lst[i], end="")
# 결과
# 123

 

리스트를 한 줄로 출력하는 경우에는 asterisk(별표)를 이용하면 더 편하다.

리스트명 앞에 *만 추가하면 예쁘게 unpacking 된다.

간편하기도 하고, 파이썬 특성상 보통 내장 기능을 사용하면 더 빠르다.

 

lst = [1, 2, 3]
print(*lst) 
# 결과
# 1 2 3

 

 

 

빠른 출력을 해야 한다면 sys모듈에 stdout.write()를 사용하면 된다.

역시 import 해야만 사용할 수 있고, print()와 달리 개행 문자가 포함되어있지 않아서 줄 바꿈을 하고 싶다면 알아서 처리해줘야 한다.

또한, 출력하고자 하는 내용을 꼭 string 형으로 넣어주어야 한다.

난 stdout.write은 거의 안 쓴다.

 

from sys import stdout

lst = [1, 2, 3]

for i in range(len(lst)):
	stdout.write(str(lst[i]))
# 결과
# 123
    
for i in range(len(lst)):
	stdout.write(str(lst[i]) + "\n")
# 결과
# 1
# 2
# 3

 

 

출력 관련해서는 사용할 수 있는 방식이 무궁무진하지만 이 정도로도 백준의 웬만한 문제들은 출력이 가능했다.