(Python/LV2) 영어 끝말잇기

문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/12981

프로그래머스

풀이:

def solution(n, words):
    # print(words[1][0])
    for i in range(1, len(words)):
        if words[i][0] != words[i-1][-1] or words[i] in words[:i]: # 끝말잇기가 아니거나 이미 앞에 있는경우
            return [ (i%n) + 1 , (i//n)+1 ]
    else:
        return [0, 0]

 

딕셔너리를 사용해 풀려했지만

같은 키에 다양한 밸류를 넣으려면 리스트를 이용해야해서 포기,