프로그래머스 < 영어 끝말잇기 > Python

📍문제 분석

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

 

📍코드

- set(집합)을 사용하여 이미 등장한 단어인지 쉽게 체크

def solution(n, words):
    answer = [0, 0]
    existing_words = {words[0]}
    round_num = 1

    for i in range(1, len(words)):
        if i % n == 0:
            round_num += 1

        if words[i][0] == words[i - 1][-1] and words[i] not in existing_words:
            existing_words.add(words[i])
        else:
            answer[0] = (i % n) + 1
            answer[1] = round_num
            return answer

    return answer


print(solution(2, ["hello", "one", "even", "never", "now", "world", "draw"]))  # [1, 3]