[프로그래머스][Lv.1] 나누어 떨어지는 숫자 배열

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12910

프로그래머스

 

내 풀이

def solution(arr, divisor):
    result = [x for x in arr if x % divisor == 0]
    if len(result) == 0:
        result = [-1]
    result.sort()
    return result

 

다른 풀이

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

 

다른 풀이를 통해 알 수 있는건

파이썬에서 [] 같이 빈 리스트는 False를 반환한다는 것과 그걸 통해 

return [] or [-1]

이렇게 리턴시켜 코드를 줄일 수 있다는 것이다.

 

또, 지금까지 sort()만 써왔는데 sorted()에 대해서도 활용해야겠다.

 

다른 풀이 보고나서 내 방식대로 새로 푼 풀이

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