프로그래머스 - 영어 끝말잇기(Python)

문제 설명

알고리즘 고민

어릴 때 자주하던 영어 끝말잇기 놀이와 똑같다. 앞사람이 말한 마지막 단어로 시작하지 않으면 탈락, 나왔던 단어를 다시 말하면 탈락. 만약 주어진 words에 탈락하는 경우가 없다면 [0,0]을 반환하도록 한다. 쉽게 해결하였다.

코드

def solution(n, words):
    answer = [0,0]
    stack = [words[0]]
    for i in range(1,len(words)):
        if words[i] in stack or stack[-1][-1] != words[i][0]:
            answer[0] = (i)%n + 1
            answer[1] = i//n + 1
            break
        stack.append(words[i])
    return answer