Python/Etc

[Python] COS Pro 2급 모의고사 - 개구리

Pydole 2018. 4. 7. 18:20

 

프로그래머스 COS Pro 2급 모의고사 - 개구리
https://programmers.co.kr/learn/courses/33/lessons/1859?language=python3
 
 

 
< 문제 설명 >

개구리가 일정한 간격으로 일렬로 놓여있는 징검다리를 건너려고 합니다.

image

징검다리에는 자연수가 적혀있으며, 개구리는 자신이 밟고 있는 징검다리에 적혀있는 숫자만큼 앞쪽으로 점프해야 합니다. 개구리는 현재

첫 번째 징검다리 위에 앉아있습니다.

징검다리에 적혀있는 숫자가 첫 번째부터 순서대로 들어있는 배열 stones가 매개변수로 주어질 때, 개구리가 징검다리를 모두 건너기 위해

필요한 점프 횟수를 return 하도록 solution 함수를 작성하려 합니다. 빈칸을 채워 전체 코드를 완성해주세요.

 

 

 

< 매개 변수 >

 

 징검다리에 적힌 숫자가 첫 번째 징검다리부터 순서대로 들어있는 배열 stones가 solution 함수의 매개변수로 주어집니다.

 

stones 배열의 길이는 1 이상 100 이하입니다.

 

stones의 원소(돌에 적혀있는 숫자)는 1 이상 5 이하의 자연수입니다

 

 

 

 

 

< 문제 풀이 >

 

 

def solution(stones):
    cnt = 0
    current = 0
    n = len(stones)
    while current < n:
        current += stones[current]
        cnt += 1

    return cnt

print(solution([2,5,1,3,2,1]))
------------------------------------------
3