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

어렵지는 않았는데 마지막 answer 에서 막혔었다.

 

 
 
def solution(n, words):
    answer = []
 
 
 
    stack = [ ]
 
    stack.append(words[0])
 
    for i in range(1,len(words)):
        # stack 에 있거나 끝말과 일치하지 않으면 진 것
 
        # 끝 말
        last = str(stack[-1])[-1]
        first = str(words[i])[0]
 
        if words[i] in stack or last!=first:
            # 누가 틀렸는지
            wrong = i
 
            # n*? + ! = wrong
            answer.append(wrong%n+1)
            answer.append(wrong//n+1)
 
 
            return answer
        stack.append(words[i])
 
    return [0,0]