[프로그래머스] 영어 끝말잇기

문제 주소

https://programmers.co.kr/learn/courses/30/lessons/12981?language=python3

코드

def solution(n, words):
    answer = []

    # [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
    print('Hello Python')
    for i in range(len(words)):
        words[i] = tuple(words[i])
    print(words)
    s = set()
    s.add(words[0])
    for i in range(1, len(words)):
        if(words[i-1][-1] != words[i][0] or words[i] in s):
            answer.append(i%n+1)
            answer.append((i//n)+1)
            print("문제 : ",i)
            break
        s.add(words[i])
    print(s)
    if(len(answer) ==0):
        answer.append(0)
        answer.append(0)
    return answer