알고리즘/프로그래머스

[Python] 프로그래머스(Lv.1) - x만큼 간격이 있는 n개의 숫자

번잔중 2022. 6. 25. 23:40

문제 설명

함수 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]

 

def solution(x, n):
    answer = []
    tmp = x
    for i in range(n):
        answer.append(x)
        x += tmp
    return answer

단순한 연산 문제입니다. 

x에 x를 n회까지 반복해서 더한 결과를 보여줍니다!

문제 풀고 다른분 코드를 봤는데 경이로워서 챙겨왔습니다.

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

감탄....!

 

피드백은 언제나 환영합니다.