[파이썬 문제] 나누어 떨어지는 숫자 배열

문제 설명

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]

 

답안

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

 

1. arr, divisor을 매개변수 삼은 solution 함수 정의

2. answer 는 리스트 [ ] 로 정의

3. arr 의 모든 요소들을 돌려보기 위해 for문을 사용

4. 만약 arr의 e요소가 divisor(자연수)로 나누었을 때 나누어 떨어진다면(0) 리스트 answer에 값을 넣어줌

5. answer값을 오름차순 정렬

6. 만약 answer 안의 리스트가 맞다면 answer을 리턴

8. 그게 아니라면 (나누어 떨어지는 element가 없다면) -1을 반환

 

---

맨 처음 문제를 읽자마자 문자를 문자로 나누라는 건가 싶었던 문제.

잘 읽어보니 자연수가 담긴 리스트의 변수였다.