문제
함수 solution은 정수 x와 자연수 n을 입력 받아, x부터 시작해 x씩 증가하는 숫자를 n개 지니는 리스트를 리턴해야 합니다. 다음 제한 조건을 보고, 조건을 만족하는 함수, solution을 완성해주세요.
제한 조건
- x는 -10000000 이상, 10000000 이하인 정수입니다.
- n은 1000 이하인 자연수입니다.
풀이
def solution(x, n):
answer = []
if x > 0 :
for i in range(x, n*x+1, x):
answer.append(i)
elif x < 0 :
for i in range(x, n*x-1, x):
answer.append(i)
else :
for i in range(n):
answer.append(x)
return answer
처음엔 if문을 이용하지 않고 for문만 이용하였는데, 그렇게 하니까 x가 양수, 음수, 0일때 경우를 고려하지 않았더라고요. 그래서 if문으로 그 경우를 다 나누어서 알고리즘을 구현하였습니다.
Uploaded by Notion2Tistory v1.0.0