for 문 4

백준 10871번 : for 와 if 같이 쓰는 문제

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

 

import sys

N, X = sys.stdin.readline().rstrip().split(' ')
A = sys.stdin.readline().rstrip().split(' ')
N = int(N)
X = int(X)
list = []
for a in A:
    a = int(a)
    if a < X:
        list.append(a)
print(list)

 

* 오답! -> 좀더 파이썬 함수를 활용하기

 

python map(func, iter) 함수는 리스트의 각 원소에 함수를 적용시킬 수 있게 해준다.

func = function으로, 주어진 iterable 의 각 element를 매핑하여 적용시켜준다

iter = iterable : 반복이 가능한 list나 tuple같은 자료형

map() 함수의 리턴은 map 객체이기 때문에 다시 리스트 등으로 변형시켜줘야 한다.

 

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

for a in A:
    if a < int(X):
        print(a, end=' ')

 

* 정답

print() 의 parameter 'end' 는 default로 '\n' 이다.

space 를 적용하여 print 를 할 때 마다 띄어쓰고 싶다면 end=' ' 을 사용하면 된다.