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

코딩테스트 연습

• programmers.co.kr
• 영어 끝말잇기

---

def solution(n, words):
    tmp = ''
    for i, v in enumerate(words):
        if tmp == '':
            tmp = v
            continue
        if tmp[-1] != v[0] or v in words[:i]:
            return [i%n+1, i//n+1]
        else:
            tmp = v
    else:
            return [0,0]