[PYTHON] 10871 X보다 작은 수 list comprehension

 

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

10871번: X보다 작은 수

문제 풀이

• X와 수열 A를 입력받아 for문을 돌린다.
• 만약 수열 A의 요소가 X보다 작다면 빈 배열에 담는다.
• 배열을 언패킹(*)해주어 출력한다.

solution

import sys
_, X = map(int, sys.stdin.readline().strip().split())
A_list = list(map(int, sys.stdin.readline().strip().split()))

result = []
for num in A_list :
  if num < X :
    result.append(num)
print(*result)

solution - list comprehension으로 풀기

import sys
_, X = map(int, sys.stdin.readline().strip().split())
A_list = list(map(int, sys.stdin.readline().strip().split()))

print(*[num for num in A_list if num < X])

2023.01.14 - [개발언어/python] - comprehension

comprehension