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

프로그래머스

 

✏️ 내가 작성한 코드

def solution(n, words):
    answer = [0,0]

    for idx,word in enumerate(words):
        if idx < 1:
            pass
        else:
            if (word in words[:idx]) or (word[0] != words[idx-1][-1]): # 전에 등장한 단어, 이전단어 끝자리로 시작 안하는 단어인 경우
                answer[0] = (idx%n)+1
                answer[1] = (idx//n)+1
                break
    return answer