[codility]FrogRiverOne

 

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

FrogRiverOne coding task - Learn to Code - Codility

오오 갓갓 코딜리티

테스트 케이스를 미리 생각하고 푸니 덜 틀리게 되었다.

A의 길이가 X보다 작으면 아무리해도 강을 못건너서 -1을 리턴하고

처음 나오는 수는 d에 저장하고 그 개수를 센다

개수가 X와 같아지면 A의 index를 리턴함

def solution(X, A):
    # write your code in Python 3.6
    if len(A) < X:
        return -1
    
    d = {}
    count = 0
    for i, v in enumerate(A):
        if v not in d.keys():
            d[v]=1
            count +=1
        if count == X:
            #print(v,count)
            return i        
    if count != X:
        return -1