0. 문제
링크) https://programmers.co.kr/learn/courses/30/lessons/12910
코딩테스트 연습 - 나누어 떨어지는 숫자 배열
array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요. divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하�
programmers.co.kr
1. 알고리즘 계획
- for문을 통해 리스트를 만든다. (리스트 컴프리핸션 이용)
- 새로 만든 리스트에 값이 들어가 있지 않으면 -1값을 추가
2. 나의 코드
def solution(arr, divisor):
answer = [v for v in arr if (v % divisor) == 0 ]
if len(answer) == 0:
answer.append(-1)
return sorted(answer)
3. 다른 사람의 코드
def solution(arr, divisor):
return sorted([n for n in arr if n%divisor == 0]) or [-1]
- OR의 특성을 이용한 풀이
- 나누어 떨어지지 않을 경우 -1을 return하게 함으로써 if문 코드 한 줄을 줄였다.
4. 정리 및 리뷰
- OR을 이용한 풀이가 굉장히 신박했다고 생각하지만,
- 기존에 기본적인 for문을 이용했다가, 리스트 컴프리핸션을 이용해 코드를 간추렸다는 점에서 내 코드도 만족스럽다.
코드 만족도: ★★★★☆
'알고리즘_코딩테스트 > 프로그래머스_Level1' 카테고리의 다른 글
| 11. 문자열 내 마음대로 정렬하기 (0) | 2020.09.10 |
|---|---|
| 10. 두 정수 사이의 합 (0) | 2020.09.09 |
| 8. 같은 숫자는 싫어 (0) | 2020.09.09 |
| 7. 가운데 글자 가져오기 (0) | 2020.09.09 |
| 6. 2016년 (0) | 2020.09.09 |