[Python] x만큼 간격이 있는 n개의 숫자

H E E 2022. 5. 27. 16:02

728x90

x만큼 간격이 있는 n개의 숫자 ( https://programmers.co.kr/learn/courses/30/lessons/12954 )

스스로 풀어볼 때

try:
    def solution(x, n):
        answer = [x] * n
        for i in range(1, n) :
            answer[i] = x*i+x
        return answer
    x, n = map(int, input().split(" "))
    print(solution(x, n))
    
except EOFError:
    print(end="")

채점 후 다른 코드를 참고하여 다시 수정할 때

try:
    def solution(x, n):
        return [x * i for i in range(1, n+1)]
    x, n = map(int, input().split(" "))
    print(solution(x, n))
    
except EOFError:
    print(end="")

코드엔 정답이 없지만 더 간단히 써보고 싶다.

🚨 try를 사용하지 않으면 EOFError 가 발생하니 주의!

728x90