[알고리즘] 프로그래머스 1단계 - 나누어 떨어지는 숫자배열

minidoo·2020년 10월 4일
0

알고리즘

목록 보기
35/85
post-thumbnail

방법1

def solution(arr, divisor):
    
    array = list(filter(lambda x:x%divisor==0, arr))
    
    if len(array) == 0:
        return [-1]
    
    array = sorted(array)
    return array
  • lambda 함수의 filter을 이용한다.


방법2

def solution(arr, divisor):
    
    answer = []
    
    for a in arr:
        if a % divisor == 0:
            answer.append(a)
    
    if len(answer) == 0:
        return [-1]
    
    return sorted(answer)
  • for 문을 돌면서 나눠 떨어지는 값을 answer에 넣는다.
  • [방법2]가 조금 더 빠르다. 쉽게 풀리는 문제의 경우, 굳이 함수를 쓰지 않아도 될 것 같다.

0개의 댓글