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

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/12910

프로그래머스

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

문제 설명

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]
입출력 예 설명
입출력 예#1
arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.

입출력 예#2
arr의 모든 원소는 1으로 나누어 떨어집니다. 원소를 오름차순으로 정렬해 [1, 2, 3, 36]을 리턴합니다.

입출력 예#3
3, 2, 6은 10으로 나누어 떨어지지 않습니다. 나누어 떨어지는 원소가 없으므로 [-1]을 리턴합니다.

 

My Sol

'''
for 문 돌면서
element % divisor == 0 
인 경우 lst에 추가
lst.sort 반환


'''

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

 

ChatGPT Sol

def solution(arr, divisor):
    lst = [i for i in arr if i % divisor == 0]  # 리스트 컴프리헨션을 사용하여 조건에 맞는 요소들을 리스트에 추가
    
    if not lst:  # lst가 비어 있다면
        return [-1]  # [-1]을 바로 반환

    lst.sort()  # 조건을 만족하는 요소가 있을 경우만 정렬
    return lst  # 정렬된 리스트 반환

- 기본적으로 리스트 컴프리헨션을 잘 안쓰다보니 코드가 길어질 수 밖에 없었음

- 조건문에서 답이 나오면 바로 return 해버리기 

 

 

정리

• 리스트 오름차순 정렬

lst.sort()

• 리스트 내림차순 정렬

lst.sort(reverse=True)

• 리스트 컴프리헨션

 lst = [i for i in arr if i % divisor == 0]  # 리스트 컴프리헨션을 사용하여 조건에 맞는 요소들을 리스트에 추가

 

• 특정 케이스일 경우 return 값이 간단하면 바로 return 해버리기

if not lst:  # lst가 비어 있다면
        return [-1]  # [-1]을 바로 반환