멀리 뛰기

멀리 뛰기

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 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 이하인 정수입니다.

입출력 예

n	result
4	5
3	3

입출력 예 설명

입출력 예 #1

위에서 설명한 내용과 같습니다.

입출력 예 #2

(2칸, 1칸)
(1칸, 2칸)
(1칸, 1칸, 1칸)
총 3가지 방법으로 멀리 뛸 수 있습니다.

---

🥨 나의 풀이

테케 1,2까지는 맞았는데 그 후로 시간초과

from itertools import product

def solution(n):
    answer = 0
    #1과 2로 만들 수 있는 길이가 n인 모든 조합을 출력하는 함수
    # 리스트 요소 값의 합이 n -> if sum(arr[i]) == n 
    #
    for i in range(1, n +1):
        elements = [1, 2]
        combinations = list(product(elements, repeat=i))
        for tpl in combinations:
            if sum(tpl) == n:
                answer += 1
              
    return answer

🌿 정답 풀이

계단을 1,2칸만 올라갈 수 있으므로 3칸을 오를 때는 1,2칸을 오르는 경우의 수를 합해주면 되고, 점화식으로 나타내면 f(n)=f(n-1)+f(n-2) => 피보나치 수열

def solution(n):
    if n<3:
        return n
    d=[0]*(n+1)
    d[1]=1
    d[2]=2
    for i in range(3,n+1):
        d[i]=d[i-1]+d[i-2]
    return d[n]%1234567  

---

itertools

반복 가능한(iterable) 객체들에 대한 효율적인 반복 작업을 수행하기 위한 도구들을 제공하는 모듈

product

• 주어진 iterable들의 데카르트 곱(Cartesian product)을 생성하는 기능을 제공
• product 함수는 두 개 이상의 iterable을 인수로 받아, 각 iterable에서 하나씩 요소를 선택하여 가능한 모든 조합을 생성
• 주어진 요소들로 가능한 모든 조합을 생성하거나, 주어진 길이의 모든 조합을 생성

  product([1, 2], ['A', 'B'])는 (1, 'A'), (1, 'B'), (2, 'A'), (2, 'B')