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

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/12981
• 끝 글자와 첫 글자가 같은지, 이미 나왔던 단어가 아닌지만 검사하면 어려울 것 없는 문제

 

def solution(n, words):
    체크 = [words[0]]   # 제일 처음 단어

    끝글자 = words[0][-1]   # 처음 1번이 말한 단어의 끝 글자
    순서 = 2    # 처음 1번은 무조건 통과라서 2번부터 시작 
    턴 = 1

    for i in range(1, len(words)):
        # 말한 단어의 앞 글자와 전에 단어의 끝 글자가 같고 말하지 않은 단어일 때
        if words[i][0] == 끝글자 and words[i] not in 체크:
            체크.append(words[i])   # 체크 배열에 추가
            끝글자 = words[i][-1]   # 현재 말한 단어의 끝 글자로 업데이트
        else:
            return [순서, 턴]       # 틀리면 현재 순서와 턴을 리턴
        if 순서 < n:    # 순서가 사람 수 보다 작으면 1씩 증가
            순서 += 1
        else:           # 순서가 사람 수 이상이 되면 1번부터 다시 하고 턴 증가
            순서 = 1
            턴 += 1

    return [0, 0]