[프로그래머스] 영어 끝말잇기 (파이썬)

dictionary 사용

def solution(n, words):
    answer = []
    dic={}
    prev=words[0]
    dic[prev]=0
    for i,w in enumerate(words):
        if i==0: continue
        if w[0]!=prev[-1] or w in dic:
            return [i%n+1,i//n+1]
        else:
            prev=w
            dic[w]=i

    return [0,0]