21. x만큼 간격이 있는 n개의 숫자 (리스트 컴프리헨션)

유형 : 구현

풀이 시간 : 5분

 

def solution(x, n):
    answer = []
    cnt = 0
    curr = x
    while cnt < n:
        cnt += 1
        answer.append(curr)
        curr += x
    return answer

 

최종 정답 코드

처음에는 for문을 사용했는데 왠지 tc 10에서 오류가 나서 while문으로 방향을 바꾸었다. 

 

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

 

리스트 컴프리헨션을 사용하는 아주아주 쉬운 풀이가 있었다.. 

 

응용하여 더 간단한 코드를 작성해 보았다,

 

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

 

파이썬의 리스트 컴프리헨션을 적극적으로 활용하자.