Python - For문 - 백준 알고리즘
Updated:
백준 알고리즘 for문 파이썬 풀이입니다.
2739번 구구단
문제
N을 입력받은 뒤, 구구단 N단을 출력하는 프로그램을 작성하시오. 출력 형식에 맞춰서 출력하면 된다.
입력
첫째 줄에 N이 주어진다. N은 1보다 크거나 같고, 9보다 작거나 같다.
2
출력
출력형식과 같게 N1부터 N9까지 출력한다.
2 * 1 = 2
2 * 2 = 4
2 * 3 = 6
2 * 4 = 8
2 * 5 = 10
2 * 6 = 12
2 * 7 = 14
2 * 8 = 16
2 * 9 = 18
정답
N = int(input())
for i in range(1,10):
print(N,'*',i,'=',N*i)
10950번 A+B-3
문제
두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)
5
1 1
2 3
3 4
9 8
5 2
출력
각 테스트 케이스마다 A+B를 출력한다.
2
5
7
17
7
정답
N = int(input())
for i in range(0,N):
a, b = input('').split()
a = int(a)
b = int(b)
print(a + b)
8393번 합
문제
n이 주어졌을 때, 1부터 n까지 합을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 n (1 ≤ n ≤ 10,000)이 주어진다.
3
출력
1부터 n까지 합을 출력한다.
6
정답
n = int(input(''))
sum = 0
for i in range(1,n+1):
sum = sum + i
print(sum)
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()을 추가로 해 주는 것이 좋다.
또한 입력과 출력 스트림은 별개이므로, 테스트케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다. 테스트케이스를 하나 받은 뒤 하나 출력해도 된다.
입력
첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.
5
1 1
12 34
5 500
40 60
1000 1000
출력
각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.
2
46
505
100
2000
정답
import sys
N = int(sys.stdin.readline())
for i in range(0,N):
A, B = sys.stdin.readline().split()
A = int(A)
B = int(B)
print(A + B)
2741번 N 찍기
문제
자연수 N이 주어졌을 때, 1부터 N까지 한 줄에 하나씩 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 100,000보다 작거나 같은 자연수 N이 주어진다.
5
출력
첫째 줄부터 N번째 줄 까지 차례대로 출력한다.
1
2
3
4
5
정답
N = int(input())
num = 1
for i in range(0,N):
print(num)
num = num + 1
2742번 기찍 N
문제
자연수 N이 주어졌을 때, N부터 1까지 한 줄에 하나씩 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 100,000보다 작거나 같은 자연수 N이 주어진다.
5
출력
첫째 줄부터 N번째 줄 까지 차례대로 출력한다.
5
4
3
2
1
정답
N = int(input())
num = N
for i in range(0,N):
print(num)
num = num - 1
11021번 A+B-7
문제
두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)
5
1 1
2 3
3 4
9 8
5 2
출력
각 테스트 케이스마다 “Case #x: “를 출력한 다음, A+B를 출력한다. 테스트 케이스 번호는 1부터 시작한다.
Case #1: 2
Case #2: 5
Case #3: 7
Case #4: 17
Case #5: 7
정답
T = int(input())
for i in range(0, T):
A , B = input().split()
A = int(A)
B = int(B)
print('Case #'+str(i+1)+':',A+B)
11022번 A+B-8
문제
두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)
5
1 1
2 3
3 4
9 8
5 2
출력
각 테스트 케이스마다 “Case #x: A + B = C” 형식으로 출력한다. x는 테스트 케이스 번호이고 1부터 시작하며, C는 A+B이다.
Case #1: 1 + 1 = 2
Case #2: 2 + 3 = 5
Case #3: 3 + 4 = 7
Case #4: 9 + 8 = 17
Case #5: 5 + 2 = 7
정답
T = int(input())
for i in range(0, T):
A , B = input().split()
A = int(A)
B = int(B)
print('Case #'+str(i+1)+':',A,'+',B,'=',A+B)
2438번 별 찍기 - 1
문제
첫째 줄에는 별 1개, 둘째 줄에는 별 2개, N번째 줄에는 별 N개를 찍는 문제
입력
첫째 줄에 N(1 ≤ N ≤ 100)이 주어진다.
5
출력
첫째 줄부터 N번째 줄까지 차례대로 별을 출력한다.
*
**
***
****
*****
정답
N = int(input())
n = 1
for i in range(N):
print('*'*n)
n = n + 1
2439번 별 찍기 - 2
문제
첫째 줄에는 별 1개, 둘째 줄에는 별 2개, N번째 줄에는 별 N개를 찍는 문제
하지만, 오른쪽을 기준으로 정렬한 별(예제 참고)을 출력하시오.
입력
첫째 줄에 N(1 ≤ N ≤ 100)이 주어진다.
5
출력
첫째 줄부터 N번째 줄까지 차례대로 별을 출력한다.
*
**
***
****
*****
정답
N = int(input())
for i in range(N):
print(' '*(N-i-1)+'*'*(i+1))
10871번 X보다 작은 수
문제
정수 N개로 이루어진 수열 A와 정수 X가 주어진다. 이때, A에서 X보다 작은 수를 모두 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 N과 X가 주어진다. (1 ≤ N, X ≤ 10,000)
둘째 줄에 수열 A를 이루는 정수 N개가 주어진다. 주어지는 정수는 모두 1보다 크거나 같고, 10,000보다 작거나 같은 정수이다.
10 5
1 10 4 9 2 3 8 5 7 6
출력
X보다 작은 수를 입력받은 순서대로 공백으로 구분해 출력한다. X보다 작은 수는 적어도 하나 존재한다.
1 4 2 3
정답
N, X = input().split()
N = int(N)
X = int(X)
List = list(map(int,input().split()))
for i in range(N):
if List[i] < X:
print(List[i],end=' ')
리스트에 map을 사용하는 것이 핵심입니다.
map(f, iterable)은 함수(f)와 반복 가능한(iterable) 자료형을 입력으로 받으며, map은 입력받은 자료형의 각 요소를 함수 f가 수행한 결과를 묶어서 돌려주는 함수입니다.
N,X를 설정하는 부분도 N,X=map(int,input().split())으로 한다면 한 줄로 정리할 수 있습니다.
Leave a comment