Lesson 4: Counting Elements → Frog River One

문제

https://app.codility.com/programmers/lessons/4-counting_elements/frog_river_one/

개구리가 가고자 하는 위치가 X라면, X에 도달하기 위해 필요한 모든 구간을 거쳐야 한다.

문제 풀이

카운트와 체크 리스트를 활용하여서 문제를 풀면 쉽게 풀 수 있다.

코드

def solution(X, A):
    check = [False] * X
    cnt = 0

    for i in range(len(A)):
        if not check[A[i] - 1]:
            check[A[i] - 1] = True
            cnt += 1
            if cnt == X:
                return i
    return -1

• 지나온 구간을 확인하기 위해 체크리스틀 통해 확인한다.
• 카운트를 활용해 카운트 값이 가고자 하는 위치와 같아지면, 인덱스 값을 반환한다.
• 반복문이 끝까지 실행된다면, 답을 찾을 수 없는 경우이다.