코딩테스트/프로그래머스

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

박소민 2022. 3. 5. 16:22
문제) 나누어 떨어지는 숫자 배열
 

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

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

programmers.co.kr

 

  • 내 풀이

- 리스트가 비어있는지 확인하는 True/False 를 이용한 if문 작성

- 리스트 내 if-else문에 -1 을 넣으면 -1이 여러개 들어갈 수 있기 때문에 밖에서 따로 정의

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

 

  • 다른 사람 풀이

- if 문 없이 [-1] 바로 넣는 법 : or [-1]

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