알고리즘 [파이썬] - 입력값을 받는 여러 가지 방법

파이썬으로 알고리즘을 시작하는 첫 번째 단계입니다. 문제를 풀기 위해서는 입력값을 적절히 받을 줄 알아야 합니다. 상황에 따라 입력받는 방법이 다르니 적절한 방법을 찾아 풀이를 시작해 보세요. 파이썬의 입력은 readline()과 input()을 사용하며, 문자열 형태로 받아옵니다.

 

입력 빠르게 받는 방법

import sys
input = sys.stdin.readline

N = input()

 

한 줄 입력받을 때

S = input().strip() # 문자열 : rstrip()이나 strip()으로 개행문자가 포함되지 않도록 공백을 제거해주는 것이 좋습니다.
N = int(input()) # int형

 

- 공백으로 구분되어 있는 입력일 때

예) 1 2 3

# 문자열
A, B, C = input().split()

# 모든 입력값을 int형으로 형변환하고 싶을 때 map() 함수를 사용
N, M, L = map(int, input().split())

 

 

- 구분자로 구분되어 있는 입력일 때

예) 1,2,3

split()의 인자로 구분자를 넣어줍니다.

# 문자열
A, B, C = input().split(",")

# 모든 입력값을 int형으로 형변환하고 싶을 때 map() 함수를 사용
N, M, L = map(int, input().split(","))

 

- 구분된 입력을 리스트로 받아오고 싶을 때 

# list로 감싸주기
str_list = list(input().split())

# list로 감싸주기
int_list = list(map(int, input().split()))

 

여러 줄 입력, 몇 줄인 지 알 때

N = int(input()) # 받아올 입력값의 줄 수

for _ in range(N):
	# 이곳에서 input()을 받아서 처리

N = int(input()) # 받아올 입력값의 줄 수

sList = [input().rstrip() for _ in range(N)] # 문자열 리스트
nList = [int(input()) for _ in range(N)]  # int형 리스트

 

여러 줄 입력, 몇 줄인지 모를 때

while True:
	# 이곳에서 input()을 받아서 처리
    if ----:
    	# 조건을 걸어서 break
    	break

while True:
    try:
        # 이곳에서 input()을 받아서 처리
    except:
        break

 

리스트를 코드 형식으로 줄 때

예) [1, 2, 3, 4, 5]

리스트 형식으로 저장됩니다.

aList = eval(input())