[코딩테스트] 입력 처리 방법

June 18, 2023

입력 처리 방법

파일로 입력 받기

표준 입력을 파일/읽기로 설정

import sys

# 표준 입력을 파일로 설정
sys.stdin = open("input.txt", "r") 

한 줄을 읽어서 정수로 변환하여 입력받기

input() : 한 줄을 읽어옴
int() : 정수로 변환

N = int(input())

한 줄을 읽고 공백으로 구분된 문자를 입력받기

input().split(구분문자) : 한 줄을 읽고 구분 문자로 나누고 리스트로 입력 받음

print(input().split())
# 콘솔창에 a, b, c를 입력하면
# ['a', 'b', 'c']

한 줄을 읽고 공백으로 구분된 문자를 정수로 변환하여 입력받기

map(형식, 리스트) : 리스트에 있는 데이터를 형식에 맞게 변환
리스트에 있는 요소의 개수에 맞춰서 변수(N, M)의 개수를 설정해야 함
콘솔창에 입력을 1 2 3(3개)을 입력하고 변수를 N, M(2개)만 주면 오류가 발생함

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

1차원 리스트로 입력 받기

list() : 괄호 안의 데이터를 리스트로 묶음

arr = list(map(int, input().split()))
# 콘솔창에 1 2 3 4 5
# [1, 2, 3, 4, 5]

이어진 숫자를 한자리씩 나눠서 리스트에 저장하여 입력받기(문자열 리스트로 저장)

arr = list(input())
# 콘솔창에 12345
# ["1", "2", "3", "4", "5"]

이어진 숫자를 한자리씩 나눠서 리스트에 저장하여 입력받기(숫자형 리스트로 저장)

arr = list(map(int,input()))
# 콘솔창에 12345
# [1, 2, 3, 4, 5]

참고 #1 : https://itcrowd2016.tistory.com/81