[백준] 10871번: X보다 작은 수(Python)

문제 

 

10871번: X보다 작은 수

 

 

---

풀이

이 문제는 for문과 if문을 같이 사용할 수 있는지 확인하는 문제였다.

그러나 내가 헤멘 부분은 오히려 *입력받는 부분*이었다. 

 

 

처음 제출한 코드

N, X = input().split()
A = []

for i in range(int(N)):
  a = int(input())
  A.append(a)
    
for i in range(len(A)):
  if A[i] < int(X):
    print(A[i], end=" ")

오류는 없었지만 런타임에러가 뜬 코드이다.

역시나 input하는 과정에서 시간이 오래걸린 것으로 보인다. 아직 파이썬 문법이 익숙치 않다보니 헤멘 것 같다.

 

 

input하는 과정을 map함수를 사용하여 수정했더니 코드도 훨씬 간결해지고 런타임도 빨라졌다.

 

 

최종 코드

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

for i in range(N):
  if A[i] < X:
    print(A[i], end=" ")