[Python] 영어 끝말잇기

(문제설명)

 

(제한사항)

 

 

(풀이 설명)

(코드 - Python)

def solution(n, words):
    answer = [0, 0]
    
    human = [0 for i in range(0, n)]
    before = []
    
    who = 0
    for word in words:
        human[who] += 1
        
        if word in before or (len(before) > 0 and before[-1][-1] != word[0]):
            answer[0] = who+1
            answer[1] = human[who]
            break
        
        before.append(word)
        who += 1
        if who == n:
            who = 0

    return answer