BOJ

BOJ I/O 처리

1. N개의 정수 입력 받기
2. input() 보다 빠르게 입력 받기

1. N개의 정수 입력 받기

# 띄어쓰기로 구분된 데이터를 1~5개 이내로 입력 받을 때
A, B, C = map(int, input().split())

# 띄어쓰기로 구분된 데이터를 여러 개 입력 받을 때 (리스트에 담기)
sample = list(map(int, input().split()))

• 위 코드의 원리

  (1) input() : 사용자로부터 임의의 값을 입력 받고, 입력된 값을 문자열(str)로 인식

  N = input("나이를 입력하세요: ")
  # 나이를 입력하세요: 

  (2) int() : 입력 값을 숫자(보통 int)로 인식해서 코드에 활용하고 싶을 때, int() 로 input() 을 감싸줌

  N = int(input("나이를 입력하세요: "))
  print(N)
  
  # 나이를 입력하세요: 12
  # 12

  (3) .split() : 문자열을 띄어쓰기 단위로 나눠서 조각난 객체들이 담긴 리스트를 만듬

  a = "저는 12살 입니다."
  print(a.split())
  
  # ['저는', '12살', '입니다.']

  b = input().split()
  # 여기서 Terminal 에 '1 2 3 4' 를 입력하면
  
  print(b)
  # ['1', '2', '3', '4'] 출력

  c, d = input().split()
  # 여기서 Terminal 에 '5 6' 을 입력하면
  
  print(c)
  # '5' 출력
  
  print(d)
  # '6' 출력

  c, d = input().split()
  # 여기서 Terminal 에 '5 6' 을 입력하면
  
  x = int(c)
  y = int(d)
  
  print(x+y)
  # '11' 출력

  (4) map(한꺼번에 적용할 함수, 반복 가능한 자료형) : input().split() 이용하여 받은 문자열 c, d 를 정수형으로 변환할 때, int() 함수를 일일이 써주는 것은 번거롭고 비효율적이므로 map() 이용하여 한꺼번에 변환

  A, B = map(int, input().split())
  # 여기서 Terminal 에 '11 22' 를 입력하면 
  
  print(type(A))
  # int
  
  print(type(B))
  # int
  
  print(A+B)
  # '33' 출력

2. input() 보다 빠르게 입력 받기

# [기존코드] 띄어쓰기로 구분된 데이터를 1~5개 이내로 입력 받을 때
A, B, C = map(int, input().split())

# [기존코드] 띄어쓰기로 구분된 데이터를 여러 개 입력 받을 때 (리스트에 담기)
sample = list(map(int, input().split()))

import sys

# [신규코드] 띄어쓰기로 구분된 데이터를 1~5개 이내로 입력 받을 때
A, B, C = map(int, sys.stdin.readline().split())

# [신규코드] 띄어쓰기로 구분된 데이터를 여러 개 입력 받을 때 (리스트에 담기)
sample = list(map(int, sys.stdin.readline().split()))

# [신규코드] 2차원 리스트를 입력 받아서 matrix 객체에 담기
matrix = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]

# [신규코드] 문자열 입력 받기
sample_str = sys.stdin.readline().rstrip()

• 위 코드의 원리

  • (1) input() vs sys.stdin.readline() 차이점

    • input() 은 입력된 값의 개행 문자(\n)를 자동으로 삭제하여 리턴 (=rstrip() 의 기능)

      >>> a = input()
      apple
      >>> a
      'apple'

    • sys.stdin.readline() 은 입력된 값의 개행 문자(\n)를 포함해서 그대로 리턴

      • 따라서 개행 문자를 제거하려면 처음부터 type 을 int 로 받아서 개행 문자 없게 만들거나, 개행 문자 제거해주는 rstrip() 을 써야함

      import sys
      
      a = sys.stdin.readline()
      apple
      >>> a
      'apple\n'

      >>> a = sys.stdin.readline().rstrip()
      apple
      >>> a
      'apple'

      >>> a = int(sys.stdin.readline())
      5
      >>> a
      5