파이썬 백준 for 문제 정복하기 (문제 11개)

[print('*' * i) for i in range(1, int(input())+1)]

문제 1.(구구단, 2739번)

N을 입력받은 뒤, 구구단 N단을 출력하는 프로그램을 작성하시오. 출력 형식에 맞춰서 출력하면 된다.

a = int(input())
for N in range(1,10):
    print(a, '*', N, '=', a*N)

문제 2.(A+B -3, 10950번)

두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

첫째 줄에 테스트 케이스의 개수 T가 주어진다.

각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)

T = int(input())

for _ in range(T):
    a,b = map(int,input().split())
    print(a+b)

for 과 in 사이에 변수를 넣어서 사용할 수 있는데, 이번 문제는 테스트 케이스의 개수 T만큼 반복하는 것이 중요하고 range함수로 생성된 숫자 요소를 변수로 선언해 사용할 필요는 없다. 이런 경우 for과 in 사이에 언더바(_)로 표현하는 것이 가능하다. 

 

문제 3. (합, 8393번)

n이 주어졌을 때, 1부터 n까지 합을 구하는 프로그램을 작성하시오.

 

풀이 1.

n = int(input())

total = 0
for i in range(1,n+1):
    total += i
print(total)

핵심 - 종료 값의 경우 해당 값을 포함하지 않기 때문에 n+1을 해준다.

 

풀이 2. Comprehension 표현식

print(sum(range(1, int(input())+1)))

 

 

문제 4. (빠른 A+B, 15552번)

본격적으로 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의 기타 여러 가지 팁을 볼 수 있다.

 

풀이 1.

import sys  # sys모듈 읽어들이기

t = int(sys.stdin.readline())

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

- 핵심 : input() vs sys.stdub.readline()

input()	sys.stdin.readline()
- input은 개행 문자를 받지 않는다. (The input takes input from the user but does not read escape character) - 입력을 받기 전에 프롬포트를 띄운다. (It has a prompt that represents the default value before the user input)	- readline은 개행문자의 입력도 받는다. (The readline() also takes input from the user but also reads the escape character) - parameter로 사이즈를 받아서 읽을 byte수를 지정할 수 있다. (음수가 아닌 수) (Readline has a parameter named size, Which is a non-negative number, it actually defines the bytes to be read)

import sys

# 여러 줄을 입력 받을 때
words = [sys.stdin.readline() for i in range(n)]

# 여러 문자를 int형으로 바로 저장하고 싶을 때
a, b = map(int, sys.stdin.readline().split())

# 여러 문자를 int형을 담은 list로 저장하고 싶을 때
word_list = list(map(int, sys.stdin.readline().split()))

풀이 2.

for l in[*open(0)][1:]:print(sum(map(int,l.split())))

- 핵심 : 백준에 올라와 있는 가장 짧은 코드이다. 코드에 대한 해석은 아직 잘 모르겠다.

 

문제 5. (N찍기, 2741번)

자연수 N이 주어졌을 때, 1부터 N까지 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

 

풀이 1.

n = int(input())

for i in range(n):
    print(i+1)

풀이 2.

n = int(input())

for i in range(1, n+1):  # 1부터 n까지
    print(i)

풀이 3.

comprehension 표현식

[print(i) for i in range(1, int(input())+1)]

문제 6. (기찍 N, 2742번)

풀이 1.

n = int(input())

for i in range(n, 0,-1):
    print(i)

핵심 : -1은 역순을 의미한다. 

range 함수의 특징 중에는 stop 값은 숫자 범위에 폼함하지 않는다.

그래서 두번 째 숫자가 0이 되어야 한다.

 

풀이2.

n = int(input())
for i in range(1, n+1)[::-1]:
    print(i)

풀이 3. 

comprehension 코드

[print(i) for i in range(1, int(input())+1)[::-1]]

 

 

문제 7. (A+B - 7, 11021번)

두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다.

각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)

출력

각 테스트 케이스마다 "Case #x: "를 출력한 다음, A+B를 출력한다. 테스트 케이스 번호는 1부터 시작한다.

T = int(input())

for i in range(1,T+1):
    a,b = map(int, input().split())
    print("Case #"+str(i)+':',a+b)

문제 8. (A+B - 8, 11022번)

두 정수 A와 B를 입력받은 다음, A+B를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다.

각 테스트 케이스는 한 줄로 이루어져 있으며, 각 줄에 A와 B가 주어진다. (0 < A, B < 10)

출력

각 테스트 케이스마다 "Case #x: A + B = C" 형식으로 출력한다. x는 테스트 케이스 번호이고 1부터 시작하며, C는 A+B이다.

T = int(input())

for i in range(1,T+1):
    a,b = map(int, input().split())
    0 < a and 10 > b
    c=a+b
    
    print("Case #"+str(i)+':',a,'+',b,'=',c)

- 핵심 : 띄어쓰기 부분은 (,)를 이용해서 한칸씩 띄어서 출력하였다. 여기서 +와 ,의 기능을 알 필요가 있는데 아래의 표에 간단하게 정리를 해보겠다.

+(plus)	,(comma)
- print 함수에서 어떤 값들을 이을 때 사용 - 공백이 추가되지 않음 >>> print("안녕" + "하세요") 안녕하세요 - +(덧셈)은 문자열들만 연결할 수 있다. >>> month = 9 >>> print(month, "월") Type Error	- print 함수에서 어떤 값들을 이을 때 사용 - 공백이 추가됨 >>>print("안녕"+"하세요") 안녕 하세요 - ,(콤마)는 모든 자료형의 값을 연결할 수 있다. >>> month = 9 >>> print(month, "월") 9월

문제 9. (별찍기 , 2438번)

첫째 줄에는 별 1개, 둘째 줄에는 별 2개, N번째 줄에는 별 N개를 찍는 문제

 

풀이 1. 

n = int(input())
for i in range(1,n+1):
    print("*" * i)

- 핵심 : print 함수 내에서 *는 문자열을 반복하는 연산자이다.

 

풀이 2. Comprehension 표현식

 

[print('*' * i) for i in range(1, int(input())+1)]

문제 10. (별찍기2, 2439번)

첫째 줄에는 별 1개, 둘째 줄에는 별 2개, N번째 줄에는 별 N개를 찍는 문제

하지만, 오른쪽을 기준으로 정렬한 별(예제 참고)을 출력하시오.

n = int(input())

for i in range(1,n+1):
    print(' '*(n-i)+'*'*i)

- 핵심 :  ' ' 여백의 빈칸을 *(반복) n-1만큼 반복하고 옆에는 이어서 +(문자열이어줌) '*'를 i만큼 출력한다.

 

문제 11. (X보다 작은 수, 10871번)

정수 N개로 이루어진 수열 A와 정수 X가 주어진다. 이때, A에서 X보다 작은 수를 모두 출력하는 프로그램을 작성하시오.

N, X = map(int, input().split())

A = list(map(int,input().split()))

if len(A) == N:
    for i in A:
        if i < X:
            print(i, end = ' ')