10871번: X보다 작은 수 - Python

https://www.acmicpc.net/problem/10871

문제

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

입력

첫째 줄에 N과 X가 주어진다. (1 ≤ N, X ≤ 10,000)

둘째 줄에 수열 A를 이루는 정수 N개가 주어진다. 주어지는 정수는 모두 1보다 크거나 같고, 10,000보다 작거나 같은 정수이다.

출력

X보다 작은 수를 입력받은 순서대로 공백으로 구분해 출력한다. X보다 작은 수는 적어도 하나 존재한다.

풀이 1

import sys
input = sys.stdin.readline

n, x = map(int, input().split())
a = map(int, input().split())
for i in (a):
    if i < x:
        print(i, end=' ')

이렇게 풀면 n이 사용되지 않는다.

n을 사용하고 싶다면,

a = list(map(int, input().split())
...
for i in range(n):
	if a[i] < x: 
    ...

와 같이 풀면 된다.