[Programmers] 나누어 떨어지는 숫자 배열

나누어 떨어지는 숫자 배열

연습 문제

 

python3

def solution(arr, divisor):
    answer = []
    
    for e in arr:
        if e % divisor == 0:
            answer.append(e)        
    
    if len(answer) == 0:
        return [-1]
    
    answer.sort();
    
    return answer

정렬을 어떻게 해야 할까 하다가.. 그냥 내장 함수 썼는데 통과.

사실 효율성 체크가 없었기도 하고..ㅋㅋ;

def solution(arr, divisor): 
	return sorted([n for n in arr if n%divisor == 0]) or [-1]

array comprehension을 이용하면 위와 같이 할 수 있다.

볼때마다 신기한 문법...