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

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

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

스택을 통해 구현했다.

나의 풀이

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