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

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

 

👩🏻‍💻 코드

def solution(n, words):
    word_set = {words[0]}
    n_cnt = [0] * (n + 1)
    n_cnt[1] += 1

    for i in range(1, len(words)):
        num = i % n + 1

        if words[i][0] != words[i - 1][-1] or words[i] in word_set:
            return [num, n_cnt[num] + 1]
        else:
            word_set.add(words[i])
            n_cnt[num] += 1

    return [0, 0]

 

📝 정리

• words를 확인하며 나오는 단어들을 word_set에 넣어준다.
• 전 사람의 마지막 글자와 같지 않거나, word_set에 존재하는 단어면 return
• 마지막까지 탈락자가 생기지 않으면 [0,0] return