Programmers - Lv2 - 멀리 뛰기

문제 설명

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 1칸, 또는 2칸을 뛸 수 있습니다. 칸이 총 4개 있을 때, 효진이는
(1칸, 1칸, 1칸, 1칸)
(1칸, 2칸, 1칸)
(1칸, 1칸, 2칸)
(2칸, 1칸, 1칸)
(2칸, 2칸)
의 5가지 방법으로 맨 끝 칸에 도달할 수 있습니다. 멀리뛰기에 사용될 칸의 수 n이 주어질 때, 효진이가 끝에 도달하는 방법이 몇 가지인지 알아내, 여기에 1234567를 나눈 나머지를 리턴하는 함수, solution을 완성하세요. 예를 들어 4가 입력된다면, 5를 return하면 됩니다.

제한 사항

n은 1 이상, 2000 이하인 정수입니다.

python code

• 팩토리얼을 이용하여 경우의 수를 계산한다.
• overflow문제가 발생하는 코드

from math import factorial


def solution(n):
    if n == 1 or n == 0:
        return n
    answer = 0
    one_step, two_step = n, 0

    while one_step != 0:
        if two_step == 0:
            answer += 1
        else:
            answer += (int(factorial(one_step + two_step) / (factorial(one_step) * factorial(two_step)))) % 1234567

        if one_step >= 2:
            one_step -= 2
            two_step += 1

        if one_step == 0:
            answer += 1
        elif one_step == 1:
            answer += (int(factorial(one_step + two_step) / (factorial(one_step) * factorial(two_step)))) % 1234567
            break

    answer %= 1234567
    return answer