[프로그래머스/Programmers] 영어 끝말잇기 - Python

Link

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

 

Solution

def solution(n, words):
    for i in range(1, len(words)):
        if words[i] in words[:i] or words[i - 1][-1] != words[i][0]:
            return [i % n + 1, i // n + 1]
    else:
        return [0, 0]

🔑 key point: return [i % n + 1, i // n + 1]