[BAEKJOON] 백준 Python 3단계 반복문 15552번 / 2741번 / 2742번 / 빠른 A+B / N찍기 / 기찍N
15552번 (빠른 A+B)
본격적으로 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의 기타 여러 가지 팁을 볼 수 있다.
입력 : 첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.
출력 : 각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.
음...ㅎ
문제가 길어서 당황했는데, python부분(파란글씨)만 보면 되서 매우 다행이다.
비슷한 문제였던 10950번은 다음과 같이 풀었다.
1. input으로 T받기
2. T 번 돌릴 수 있는 for문 만들기
3. A,B input 받기
4. A+B 출력하기.
T = int(input())
for i in range(T):
A,B = map(int, input().split())
print(A+B)
하지만 이번 문제에서는
조건문(for) 안에서는 input 대신 sys.stdin.readline을 사용하고,
sys.stdin.readline을 사용하면 맨 끝의 개행문자까지 같이 입력을 받아
문자열로 저장을 해야하는데 .rstrip()을 추가하면 해결이 된다고 한다.
일단 써있는대로 해보자.!
import sys
T = int(input())
for i in range(T):
A,B = map(int, sys.stdin.readline().split())
print(A+B)
먼저 sys모듈을 import 시켜주고
A,B 인풋 할 때 input()이 아닌 sys.stdin.readline()을 넣었더니 정답!
그리고,
.rstrip()도 궁금해 넣어보았다.
import sys
T = int(input())
for i in range(T):
A,B = map(int, sys.stdin.readline().rstrip().split())
print(A+B)
문자열로 저장이 되는 것같은데 이렇게 해도 정답처리가 되었다.
2741번 (N찍기)
자연수 N이 주어졌을 때, 1부터 N까지 한 줄에 하나씩 출력하는 프로그램을 작성하시오.
입력 : 첫째 줄에 100,000보다 작거나 같은 자연수 N이 주어진다.
출력 : 첫째 줄부터 N번째 줄 까지 차례대로 출력한다.
1. 자연수 N을 input()한다.
2. N만큼 돌아가는 for문 작성한다.
3. x에 1씩 더해주는 것을 N만큼 한다.
4. x를 출력한다.
N = int(input())
x =0
for i in range(N):
x += 1
print(x)
사실 식을 차근히 세웠다기 보다는,
그냥 이렇게 하면 될 것같은데? 하고 막 쳤더니 되었다.!ㅎㅎ
2742번(기찍 N)
자연수 N이 주어졌을 때, N부터 1까지 한 줄에 하나씩 출력하는 프로그램을 작성하시오.
입력 : 첫째 줄에 100,000보다 작거나 같은 자연수 N이 주어진다.
출력 : 첫째 줄부터 N번째 줄 까지 차례대로 출력한다.
문제이름이 기찍 N 이길래 뭔소리지 했는데
위에 문제를 거꾸로 출력하는 것이였다ㅋㅋㅋ
백준에 은근 이런 개그코드?가 있는 것같다
1. N을 input해준다.
2. x에 N+1을 할당해준다.
3. N번 돌아가는 for문을 작성해준다.
4. x에서 -1씩 차례대로 빠지게 해준다.
5. x를 출력한다.
N = int(input())
x = N+1
for i in range(N):
x -= 1
print(x)