[프로그래머스] 파이썬 - 영어 끝말잇기 Lv.2

Python : 영어 끝말잇기 (프로그래머스)

 

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

 

def solution(n, words):
    checks_word = []
    answer=[0,0]
    checks_word.append(words[0])

    for i in range(1, len(words)):
        if words[i] not in checks_word and words[i-1][-1] == words[i][0]:
            checks_word.append(words[i])
        else:
            answer[0] = (i %  n) + 1
            answer[1] = (i // n) + 1
            break
    return answer


ans = []
words=["tank", "kick", "know", "wheel", "land", "dream", "mother", "robot", "tank"]
n=3
ans = solution(n, words)
print(ans)

 

실행 결과

프로그래머스 실행 결과