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

문제

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.

divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

제한 조건

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

나의 풀이:

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

어려움 없이 생각나는대로 문제를 풀었다. 이제 코드를 줄여보자.

def solution(arr, divisor):
    return sorted([i for i in arr if i % divisor == 0])

깔-끔.

아, 그런데 나누어 떨어지는 수가 하나도 없으면 배열에 -1을 담아야 되는걸 깜빡했다ㅠㅠ

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

최종 결과물

다른 사람들의 풀이:

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

파이썬은 or 앞이 참일 경우 해당 값까지만, 거짓일 경우 뒤에까지 호출하기 때문에 내 코드에서 뒤에 ‘or [-1]’ 만 붙여주면 되는 것이었다…

기초적인 부분이지만 and, or 의 정의에 대해서 더 명확하게 알고 있는 것이 좋겠다.