Coding/Programmers
[코딩테스트 입문] 나누어 떨어지는 숫자 배열(Python3)
Soo_buglosschestnut
2023. 1. 3. 13:52
[코딩테스트 입문] 나누어 떨어지는 숫자 배열(Python3)
https://school.programmers.co.kr/learn/courses/30/lessons/12910?language=python3
append랑 sorted 이용해서 했다.
def solution(arr, divisor):
answer = []
for i in range(len(arr)):
if arr[i] % divisor == 0:
answer.append(arr[i])
if len(answer) == 0:
answer.append(-1)
return sorted(answer)
한줄 최고!
or 마무리 !
def solution(arr, divisor): return sorted([n for n in arr if n%divisor == 0]) or [-1]