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

문제링크

스택 활용하여 단어의 마지막 영어와 다음 단어의 첫 영어 비교

파이썬

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