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

영어 끝말잇기

문제

코드

• 조건에 만족하면 리스트에 추가(append)하는 방식으로 끝말잇기 진행
• 조건에 맞지 않는다면 중단하고 return

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

    return [0,0]