ror_coding

[Programmers] 멀리 뛰기 - 12914 본문

Algorithm/Python

[Programmers] 멀리 뛰기 - 12914

ro_rdil_31 2024. 10. 31. 23:19
728x90

math.comb()를 이용해서 구하는 코드.

 

Question

 

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

 

 

Code

 

import math
def solution(n): 
    answer = 0
    twos = n//2
    start = twos
    
    if n%2==1 : start += 1
    
    for i in range(start, n+1): 
        answer += math.comb(i,twos)
        twos -= 1
        
    return answer % 1234567

 

now me

On my github

 

728x90