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

문제

• 프로그래머스 2018 서머코딩- 영어 끝말잇기
• https://programmers.co.kr/learn/courses/30/lessons/12981

풀이

끝말잇기 탈락 조건은 잘못된 단어를 말하거나, 이전에 나왔던 단어를 말하는 경우이다.

따라서 탈락 조건에 대해 검사하고 check를 True로 바꿔준다.

check가 True라면 탈락된 것이므로 탈락된 사람의 번호와 몇 번째에탈락하는지 구하여 반환한다.

코드

def solution(n, words):
    answer = []
    check = False
    end = words[0][-1]
    for i in range(1,len(words)):
        p1 = words[i-1]
        p2 = words[i]
        for j in range(0,i):
            if words[i] == words[j]:
                check = True
                break;
        if p1[-1] != p2[0]:
            check = True
        if check:
            answer.append(i%n+1)
            answer.append(i//n+1)
            return answer
    answer.append(0)
    answer.append(0)
    return answer