boj 입력값 받기 (python)

boj 문제를 풀 때, 입력데이터를 받는 여러가지 방법이다.

1, 2번은 한 줄에 데이터를 받을 때,
3번은 여러 줄의 데이터를 받는 경우에 사용한다.

1. 값 하나를 받을 때

N = int(input())

2. 값 여러 개를 받을 때

• map 함수를 통해 입력받는 동시에 각 원소를 int로 변환한다.

a = list(map(int, input().split()))

3. 여러 줄을 받을 때

• 몇 줄인지 정해져 있을 때

import sys

data = []
for i in range(10):
  line = sys.stdin.readline().rstrip()
  data.append(int(line))

• 정해져 있지 않을 때

import sys
data = []

while True:
  try:
    line = sys.stdin.readline()
    if not line:
      break
    data.append(int(line.rstrip()))
  except:
    break

• 로컬 테스트하는 방법

f = open('input.txt', 'r')

data = []
for i in range(10):
  line = f.readline().rstrip()
  data.append(int(line))