문제
본격적으로 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()을 추가로 해 주는 것이 좋다.
또한 입력과 출력 스트림은 별개이므로, 테스트케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다. 테스트케이스를 하나 받은 뒤 하나 출력해도 된다.
자세한 설명 및 다른 언어의 경우는 이 글에 설명되어 있다.
이 블로그 글에서 BOJ의 기타 여러 가지 팁을 볼 수 있다.
학습
-input() 대신 sys.stdin.readline()을 사용하는 이유
반복적으로 여러줄을 입력받아야 할 때 input()으로 입력을 받는다면 시간초과가 발생할 수 있다.
sys.stdin.readline()는 한 줄 단위로 입력받기 때문에, 개행문자가 같이 입력 받아진다. 예를들어 3을 입력했다면 3\n까지 같이 입력받아 진다. 또한 변수 타입이 str로 저장되기 때문에 정수로 사용하기 위해서는 int 형 변환을 거쳐야 한다.
개행문자 같이 입력, int 형 변환
a,b = map(int,sys.stdin.readline().split())
코드
import sys num=int(input()) for i in range(num): a,b=map(int,sys.stdin.readline().split()) print(a+b)
https://www.acmicpc.net/problem/15552
'프로그래밍 > python' 카테고리의 다른 글
[python] 백준 5543 상근날드 (0) | 2022.01.12 |
---|---|
[python] 백준 1009 분산처리 (0) | 2022.01.11 |
[python] 백준 1330 두 수 비교하기 (0) | 2021.12.26 |
[python] 백준 2941 크로아티아 알파벳 (0) | 2021.12.26 |
[학교/파이썬/20.06.22] 과제2 (1) | 2020.06.30 |