[프로그래머스 코딩테스트 연습] 15. 나누어 떨어지는 숫자 배열

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

---

1) 문제 설명

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

 

2) 제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

 

3) 입출력 예

입출력 예#1
arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.

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

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

 

---

1. 문제 접근방식

for문을 활용하여 arr의 모든 수를 divisor로 나누어 나머지가 0인 경우를 확인해보았다.

 

2. 풀이 코드

1) 나의 풀이 코드

def solution(arr, divisor):
    answer = []
    for i in range(len(arr)):
        if arr[i] % divisor == 0:
            answer.append(arr[i])
            answer.sort()   # 오름차순으로 정렬
    if len(answer) == 0:    # if answer == []: // if not answer:
        answer.append(-1)
    return answer

arr의 길이를 range하여 arr의 인덱스로 활용한 후 divisor로 나누었을 때, 0이 나온 값을 answer 리스트에 담아준다.

answer 리스트에 아무것도 없다면 -1을 출력한다.

 

그리고 중요한 점!

sort() 함수를 사용하여 answer에 들어온 값들을 정렬해줘야 한다. (arr의 값들이 정렬되어있지 않기에 해당 작업이 필요하다)

 

2) 다른 사람 풀이 코드

def solution(arr, divisor):
    answer = []
    arr.sort()
    for i in arr:
        if i%divisor==0:
            answer.append(i)
    return answer if len(answer)!=0 else [-1]    # 해당 코드 주의깊게 보기!

마지막 return 코드를 유심히 봐야 한다.

마지막 코드는 answer의 길이가 0이 아니면 answer을 출력, 길이가 0이라면 -1을 출력해라는 의미이다.

이런 식으로도 코드를 구현할 수 있다는 것을 배울 수 있었다.

 

3. 배운 점

sort 함수가 생각보다 많이 사용하는데, sort 함수를 아직 명확하게 활용하지 못한 것이 아쉽다.

리스트.sort() 이렇게 사용된다는 것을 꼭 기억하기!

 

또한 answer 리스트에 담긴 것이 없다는 것을 if 조건문으로 조건을 걸었는데,

이때 answer 길이가 아니라 'if answer == []:', 'if not answer:' 이렇게도 코드를 짤 수 있다는 것을 알게 되었다.

다음에는 해당 코드를 활용해봐야겠다 :)