[프로그래머스] Summer/Winter Coding(~2018) 영어 끝말잇기 문제 풀이



1. 문제

https://programmers.co.kr/learn/courses/30/lessons/12981

2. 풀이

def solution(n, words):
    answer = []
    used = {words[0]: True}
    i = 1
    while i < len(words):
        # 사용 여부 체크
        if words[i] in used:
            return [i % n + 1, i // n + 1]
        used[words[i]] = True
        # 앞 단어 끝자리와 현 단어 앞자리 비교
        if words[i - 1][-1] != words[i][0]:
            return [i % n + 1, i // n + 1]
        i += 1
        
    return [0, 0]