[Python] 프로그래머스 - Level2 - 영어 끝말잇기

Summer/Winter Coding(~2018)

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

import math

def solution(n, words):
    answer = [0, 0]
    word_list = []
    w = words[0][0]
    for idx, word in enumerate(words):
        if word[0] != w or word in word_list:
            num = (idx + 1) % n
            if num == 0:
                answer[0] = n
            else:
                answer[0] = num

            answer[1] = math.ceil((idx + 1) / n)
            break
        word_list.append(word)
        w = word[-1]
    
    return answer