백준 15552 파이썬 - input를 대신할 함수가 있다고?!

백준 15552번 - 빠른 A + B with python

Q. 본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다. 입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.

C++을 사용하고 있고 cin/cout을 사용하고자 한다면, cin.tie(NULL)과 sync_with_stdio(false)를 둘 다 적용해 주고, endl 대신 개행문자(\n)를 쓰자. 단, 이렇게 하면 더 이상 scanf/printf/puts/getchar/putchar 등 C의 입출력 방식을 사용하면 안 된다.

Java를 사용하고 있다면, Scanner와 System.out.println 대신 BufferedReader와 BufferedWriter를 사용할 수 있다. BufferedWriter.flush는 맨 마지막에 한 번만 하면 된다.

Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다. 단, 이때는 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.

또한 입력과 출력 스트림은 별개이므로, 테스트케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다. 테스트케이스를 하나 받은 뒤 하나 출력해도 된다.

입력> 첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.

출력> 각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.

---

전에 포스팅했던 A+B를 기억하는가? 기억이 나질 않는다면 이 포스트를 참고하길 바란다.

2022.03.20 - [우당탕탕 파이썬 with 하루하나 알고리즘] - 백준 10950 파이썬 - for문을 활용해 A+B 구하기

백준 10950 파이썬 - for문을 활용해 A+B 구하기

이 포스팅에서는 변수를 입력받을 때 input()함수를 사용하였다. 하지만, 이 함수를 쓰면 조금 느릴 수도 있다는 단점이 있다. 그래서 입력받을 수의 개수가 많아지거나, 빅데이터를 활용하게 된다면 input대신 sys.stdin.readline을 사용해주는 것이 좋다.

이 함수를 사용해줄 때 주의할 점이 있다. 이 함수는 그냥 사용하면 안된다.

이 함수를 쓰기 전에 import sys를 통해 이와 관련된 함수를 사용할 거에요!라고 선언을 해주어야 한다.

 

이제 코드를 적용해보자.

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

for i in range(T):
    a, b = map(int, sys.stdin.readline().split())
    print(a+b)

처럼 사용해주면 된다.

---

나는 파이썬 IDE를 스파이더로 사용하고 있는데, spyder환경에서는 저 코드가 작동하지 않는다고 한다. 계속 오류나서 당황했는데 그렇다구 하더라구요.. 스파이더랑 주피터 환경에서는 사용이 안된다고 한다.

스파이더 환경에서는 이런 오류가 나니까 참고하시길!!