[Python | 프로그래머스] 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]

 

나의 풀이

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

 

파이썬 문법을 간단히 공부하고 풀어본 첫번째 문제이다.

 

 

더 간단한 풀이

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

리스트 컴프리헨션을 사용하여 훨씬 간단하게 풀 수 있다.

 

 

사용된 문법 : 리스트 컴프리헨션 활용하기

대괄호 ([]) 안에 조건문과 반복문을 넣는 방식으로 리스트를 초기화할 수 있다.

계속 사용하는 것이니 꼭 기억하자!

[expression for item in iterable]

squares = [x**2 for x in range(1, 6)]

even_numbers = [x for x in range(1, 11) if x % 2 == 0]

 

 

ref.

https://school.programmers.co.kr/learn/courses/30/lessons/12954

프로그래머스