[프로그래머스] #25 나누어떨어지는 숫자배열 (sorted(iterable)=[list])

▶ 문제

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 element in arr:
        if element % divisor == 0:
            answer.append(element)
    if answer:
        return sorted(answer)
    else:
        return [-1]

통과!

 

 

 

이걸 한줄로 적을 수도 있다

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

or 을 붙이면 앞의 표현식이 거짓일 경우 뒷부분이 반환된다.

sorted([n for n in arr if n%divisor == 0]) 의 결과값이 False일 경우,

즉 빈 리스트가 결과일 경우 [-1] 을 반환한다.

이외에도 빈 문자열, 0, None 등이 False로 취급된다.

 

 

그리고 이 코드에서는 리스트와 append()를 사용하지 않았는데,

sorted() 함수는 인자를 받아서 알아서 리스트로 반환하기 때문이다.

sorted() 함수는 인자로 리스트 뿐 아니라 iterable 객체를 받을 수 있다.

반복을 통해 생성된 값을 하나씩 가져와 정렬된 [리스트]로 반환한다.

 

 

 

---