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

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

# 조건

• 1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다.
• 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.
  1. 1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말합니다.
  2. 마지막 사람이 단어를 말한 다음에는 다시 1번부터 시작합니다.
  3. 앞 사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 합니다.
  4. 이전에 등장했던 단어는 사용할 수 없습니다.
  5. 한 글자인 단어는 인정되지 않습니다.
• 다음은 3명이 끝말잇기를 하는 상황을 나타냅니다.
  • tank → kick → know → wheel → land → dream → mother → robot → tank
• 위 끝말잇기는 다음과 같이 진행됩니다.
  • 1번 사람이 자신의 첫 번째 차례에 tank를 말합니다.
  • 2번 사람이 자신의 첫 번째 차례에 kick을 말합니다.
  • 3번 사람이 자신의 첫 번째 차례에 know를 말합니다.
  • 1번 사람이 자신의 두 번째 차례에 wheel을 말합니다.
  • (계속 진행)
• 끝말잇기를 계속 진행해 나가다 보면, 3번 사람이 자신의 세 번째 차례에 말한 tank 라는 단어는 이전에 등장했던 단어이므로 탈락하게 됩니다.
• 사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.

제한 사항

• 끝말잇기에 참여하는 사람의 수 n은 2 이상 10 이하의 자연수입니다.
• words는 끝말잇기에 사용한 단어들이 순서대로 들어있는 배열이며, 길이는 n 이상 100 이하입니다.
• 단어의 길이는 2 이상 50 이하입니다.
• 모든 단어는 알파벳 소문자로만 이루어져 있습니다.
• 끝말잇기에 사용되는 단어의 뜻(의미)은 신경 쓰지 않으셔도 됩니다.
• 정답은 [ 번호, 차례 ] 형태로 return 해주세요.
• 만약 주어진 단어들로 탈락자가 생기지 않는다면, [0, 0]을 return 해주세요.

# 접근 방법

• words를 순회하며 조건에 맞게 탈락 조건을 설정해준다.
  • 앞 단어의 마지막 글자와 현재 단어의 첫 글자가 다른 경우 탈락
  • 중복된 단어를 말하는 경우 탈락
• 첫 번째 케이스의 경우 매번 앞 단어와 비교해주면 된다.
• 두 번째 중복된 단어를 체크해주기 위해서 딕셔너리를 사용해준다.
  • dict[현재 말한 단어].get() 이 유효하지 않다면 새로 추가해주고
  • 유효하다면 이미 앞서 말했으므로 종료시켜준다.
• 출력의 경우, (현재 단어의 인덱스) % 사람 수 + 1 를 통하여 번호를 뽑아주고 (현재 단어의 인덱스) // 사람 수 +1 를 통하여 차례를 구해주면 된다.

def solution(n, words):
    answer = [0, 0]
    check = dict()
    for idx, word in enumerate(words):
        if idx == 0:
            check[word] = 1
            continue
        else:
            if word[0] != words[idx -1][-1]:
                break
            elif check.get(word):
                break
            else:
                check[word] = 1
    else:
        return [0, 0]

    answer = [idx % n + 1, idx // n + 1]
    return answer