본문 바로가기

코테

[Programmers] 영어 끝말잇기 python

반응형
def solution(n, words):
    answer = [0,0]
    word = words[0]
    dict = {}
    dict[word] = 1
    now = 2
    loop = 1
    for i in range(1,len(words)):
        if word[-1] != words[i][0]:
            return [now,loop]
        if words[i] in dict:
            return [now,loop]
        else:
            dict[words[i]] = 1
            word = words[i]
            now +=1
            if(now > n):
                now = 1
                loop += 1
    return answer
반응형