2019.05.22

프로그래머스 알고리즘 문제 (lv1)

• 나누어 떨어지는 숫자 배열

문제 설명

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에서 한개씩 뽑는다.
2. 주어진 divisor로 나누어서 나누어 떨어지면 배열에 담는다.
3. 배열이 없다면 -1을 남는다.
4. 오름차순으로 정열하여 반환한다.

컴퓨터에게는?

1. for문을 돌려서 arr에서 한개씩 가지고 온다.
2. 만약 나누어 나머지가 0이면 배열에 더한다.
3. 배열이 없다면 -1을 넣는다.
4. 배열을 정렬하여 뽑아 낸 값을 return에 담는다.

코드

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

• 포문이 들어가게 되는데 잘 짠 것인지 헷갈린다.

다른 사람 풀이

def solution(arr, divisor):
    arr = [x for x in arr if x % divisor == 0];
    arr.sort();
    return arr if len(arr) != 0 else [-1];

• arr을 한줄로 표현해 주었다.
• 또 len(arr)과 [-1]을 한 줄로 적어주었다.
• 파이토닉하넹…
• 항상 푼 이후에 줄이는 연습을 해보면 좋을 것 같다는 생각이 든다.