1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.
1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말합니다.
마지막 사람이 단어를 말한 다음에는 다시 1번부터 시작합니다.
앞사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 합니다.
이전에 등장했던 단어는 사용할 수 없습니다.
한 글자인 단어는 인정되지 않습니다.
다음은 3명이 끝말잇기를 하는 상황을 나타냅니다.
tank → kick → know → wheel → land → dream → mother → robot → tank
위 끝말잇기는 다음과 같이 진행됩니다.
끝말잇기를 계속 진행해 나가다 보면, 3번 사람이 자신의 세 번째 차례에 말한 tank 라는 단어는 이전에 등장했던 단어이므로 탈락하게 됩니다.
사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.
n words result
3 ["tank", "kick", "know", "wheel", "land", "dream", "mother", [3,3]
"robot", "tank"]
5 ["hello", "observe", "effect", "take", "either", "recognize", [0,0]
"encourage", "ensure", "establish", "hang", "gather", "refer",
"reference", "estimate", "executive"]
2 ["hello", "one", "even", "never", "now", "world", "draw"] [1,3]
Code
def solution(n, words): w_list = [] for idx in range(len(words)): if words[idx] in w_list: # 해당 단어가 이전에 나온 경우 result = divmod(idx+1, n) if result[1] == 0: return [n, result[0]] else: return [result[1], result[0]+1] if idx != len(words)-1: # 해당 단어의 시작이 이전 단어의 끝과 일치하지 않는 경우 w_list.append(words[idx]) if words[idx][-1] != words[idx+1][0]: result = divmod(idx+2, n) if result[1] == 0: return [n, result[0]] else: return [result[1], result[0]+1] return [0,0]
- 문제에서 탈락하는 경우는 해당 단어가 이전에 나왔거나 끝말잇기가 안되는 경우이므로 해당 조건을 나누어 구현하였고 해당 조건에 만족하는 경우의 index 값을 기준으로 참가자의 번호와 차례를 구했음.