[1주차] 알고리즘을 위한 파이썬 기본 문법

input()과 sys.stdin.readline() 은 동일하다.

다만, 알고리즘 문제 풀때 빠른 속도를 위해, sys.stdin.readline()을 쓸 것을 추천...!

(import sys 넣는거 깜박하지 말아주세요..!)

 

한개의 데이터 입력

import sys

T = int(sys.stdin.readline())

sys.stdin.readline()의 자료형은 str이므로, int로 바꿔주는 과정 필요

 

정해진 갯수의 데이터 입력  &  space로 구분받아 입력

import sys

A, B = sys.stdin.readline().split()
A = int(A)
B = int(B)

A, B = int(sys.stdin.readline().split()) 이렇게 한 줄에 못쓴다!

 

임의의 갯수의 데이터 입력 & space로 구분받아 입력

import sys

array = list(map(int,sys.stdin.readline().split()))

map()은 객체를 만들어 주기 때문에, list로 형변환해주는 list()가 필요

 

 

n개의 줄로 입력해 배열에 담기

import sys

array = [int(sys.stdin.readline()) for i in range(n)]

import sys
range = []
T = int(sys.stdin.readline())
for i in range(T):
    data.append(list(map(int,sys.stdin.readline().split())))

 

문자열을 n줄의 데이터 입력받아 리스트에 넣기

import sys

T = int(sys.stdin.readline())
range = [sys.stdin.readline().strip() for i in range(T)]