본문 바로가기
Engineering WIKI/Programmers

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

by wonos 2022. 4. 30.
 

코딩테스트 연습 - x만큼 간격이 있는 n개의 숫자

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

programmers.co.kr

내풀이 (성공)

def solution(x, n):
    answer = []
    if x == 0:
        answer = [0] * n
        return answer;
    if x > 0:
        repeat_range = (x * n) + 1
    if x < 0:
        repeat_range = (x * n) - 1
    for i in range(x, repeat_range, x):
        answer.append(i)
    return answer

다른방법 1

def number_generator(x, n):
    # 함수를 완성하세요
    return [i * x + x for i in range(n)]
print(number_generator(2, 5))