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

yujeongkwon·2022년 1월 4일
0
post-custom-banner

문제설명

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

제한 조건
x는 -10000000 이상, 10000000 이하인 정수입니다.
n은 1000 이하인 자연수입니다.

과거의 내가 무슨 생각을 했는지 난 잘 모르겠다. 심지어 두번째 코드가 더 빠름

파이썬 리스트 표현식( list comprehension )을 사용하면 더 간단하게 표현 가능. 나중에 따로 정리 해야겠다.

2021 8월 코드

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

2022.01.04 코드

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

리스트 표현식

def solution(x, n):
    answer = [x*i for i in range(1,n+1)]
    return answer
profile
인생 살자.
post-custom-banner

0개의 댓글