[하루한문제] 프로그래머스::영어 끝말잇기 Lv2 (w.Python)

프로그래머스

 

📌 문제 요약

- n명의 사람이 영어 끝말잇기 진행 중, 탈락자가 생긴다면 탈락자의 번호와 차례 번호를 출력하는 함수 작성
- 탈락 조건 ① 앞 사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야함
- 탈락 조건 ② 이전에 등장했던 단어를 말하면 탈락
- 탈락자가 없다면 [0,0]을 리턴
- input : n = 참가자 수, words = 끝말잇기에 사용한 단어들
- output : [탈락한 참가자번호, 게임 차례]

 

📌 입출력 예시

n	words	result
3	["tank", "kick", "know", "wheel", "land",. "dream", "mother", "robot", "tank"]	[3,3]
5	["hello", "observe", "effect", "take", "either", "recognize", "encourage", "ensure", "establish", "hang", "gather", "refer", "reference", "estimate", "executive"]	[0,0]
2	["hello", "one", "even", "never", "now", "world", "draw"]	[1,3]

 

✅ 나의  풀이

>> 필요 단계

① 게임에 참여하는 사람들의 번호를 저장하는 'p' 리스트 생성

  - 만약 n이 3이라면 p = [1, 2, 3]

② words의 길이만큼 반복문 진행

③ 이전 단어의 끝자리와 지금 단어의 앞자리가 같지 않거나 이전에 말한 단어라면 해당 순서 사람이 탈락

③ 탈락자가 없이 게임이 끝났다면 [0,0] 출력

 

import math

def solution(n, words):
    answer = []
    
    p = [i+1 for i in range(0, n)]

    for idx, word in enumerate(words):
        
        if idx > 0 and (words[idx][:1] != words[idx-1][-1] or word in words[:idx]):
            
            return [p[idx%n], math.ceil((idx+1)/n)]
            
        else:
            answer = [0,0]
        
    return answer