x만큼 간격이 있는 n개의 숫자 - Python

def solution(x, n):
    if x:
        return list(range(x, x * n + 1 if x >= 0 else x * n - 1, x))
    else:
        return list([0 for _ in range(n)])


# Test Cases
print(solution(2, 5))
print(solution(4, 3))
print(solution(-4, 2))
print(solution(0, 2))

  range()를 사용하여 풀고 싶었습니다. 그런데 range()의 세 번째 인수에는 0이 들어가면 안 되는걸 알았습니다. 그래서 if()를 사용했습니다...

 

---

# 문제가 개편되었습니다. 이로 인해 함수 구성이나 테스트케이스가 변경되어, 과거의 코드는 동작하지 않을 수 있습니다.
# 새로운 함수 구성을 적용하려면 [코드 초기화] 버튼을 누르세요. 단, [코드 초기화] 버튼을 누르면 작성 중인 코드는 사라집니다.
def number_generator(x, n):
    # 함수를 완성하세요
    return [i * x + x for i in range(n)]
print(number_generator(2, 5))

  근데 다른 사람의 풀이를 보니까 이렇게 했네요. 생각이 부족했습니다.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/12954

프로그래머스