프로그래머스 LEVEL1 - 나누어 떨어지는 숫자 배열

▶ 문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.제한사항

▶ 제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

▶ 입출력 예

 arr                               divisor    return

[5, 9, 7, 10]	5	[5, 10]
[2, 36, 1, 3]	1	[1, 2, 3, 36]
[3,2,6]	10	[-1]

▶ 내가 짠 코드

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

* return answer.sort() 는 None을 반환한다?

 - sort()는 목록을 제자리에 정렬하는 것이지 새 목록을 반환하는 것이 아니므로 반환값이 없다. 따라서 return 시 None이 반환된다.

 - sorted()는 새로운 목록을 반환하므로 이걸로 해결.

▶ 참고사항

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

 다른 사람 풀이를 참고했는데 새로운 사실을 알게 되었다.

 return에서도 or / and 연산자를 사용할 수 있다.