영어 끝말잇기

 

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

프로그래머스

문제풀이

def solution(n, words):
    answer = [0, 0]

    cnt = {i: 0 for i in range(1, n + 1)}
    unique = set()
    temp = ""
    order = 1

    for i, word in enumerate(words):
        cnt[order] += 1

        # 중복된 단어 포함, 끝말잇기가 아니면 탈락 처리
        if word in unique or i != 0 and temp != word[0]:
            answer = [order, cnt[order]]
            break

        unique.add(word)
        temp = word[-1]
        order = (order % n) + 1

    return answer