[프로그래머스] 영어 끝말잇기, Summer/Winter Coding(~2018), python

https://programmers.co.kr/learn/courses/30/lessons/12981

코딩테스트 연습 - 영어 끝말잇기

def solution(n, words):
    answer = [0, 0]
    word_set = set([words[0]])
    for i in range(1, len(words)):
        if words[i - 1][-1] != words[i][0] or words[i] in word_set:
            # print("hello", i)
            answer[0] = i % n + 1
            answer[1] = i // n + 1
            break
        word_set.add(words[i])


    return answer

끝말잇기가 되는지, 중복이 되는지만 확인하면 된다.