[프로그래머스] x만큼 간격이 있는 n개의 숫자

 

문제 설명

함수 solution은 정수 x와 자연수 n을 입력 받아, x부터 시작해 x씩 증가하는 숫자를 n개 지니는 리스트를 리턴해야 합니다. 다음 제한 조건을 보고, 조건을 만족하는 함수, solution을 완성해주세요.

제한 사항

-x는 -10000000 이상, 10000000 이하인 정수입니다.

-n은 1000 이하인 자연수입니다.

입출력 예

x	n	answer
2	5	[2,4,6,8,10]
4	3	[4,8,12]
-4	2	[-4, -8]

접근법

음수, 양수, 0일때를 다르게 처리해야 합니다. 먼저 0일때는 0씩 증가하기 때문에 모든 요소가 0이 됩니다. 즉, 0인 요소를 n개 가지면 됩니다. 양수 같은 경우에는 range(x, x*n+1)로 설정해줘야 하고, 음수 같은 경우에는 range(x, x*n-1)로 설정해줘야 합니다.

​

나의 코드

def solution(x, n):
    if x > 0:
        answer = [num for num in range(x,x*n+1,x)]
    elif x < 0:
        answer = [num for num in range(x,x*n-1,x)]
    else:
        answer = [0]*n
    return answer

다른 코드

def solution(x, n):
    return [x*i for i in range(1, n+1)]

x를 x씩 증가시키는것은 배수를 구하는것과 같습니다! 따라서 위와 같이 코드를 정의할 수도 있습니다.