[Python/파이썬] 프로그래머스 Summer/Winter Coding(~2018) 영어 끝말잇기

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

프로그래머스

코드

from collections import defaultdict

def solution(n, words):
    answer = [0, 0]
    word_dict = defaultdict(list)
    word_dict[words[0][0]].append(words[0])
    
    for idx in range(1, len(words)):
        if words[idx] in word_dict[words[idx][0]]:
            return [idx % n + 1, idx // n + 1]
        if words[idx][0] != words[idx-1][-1]:
            return [idx % n + 1, idx // n + 1]
        word_dict[words[idx][0]].append(words[idx])

    return answer