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

https://school.programmers.co.kr/learn/courses/30/lessons/12981

프로그래머스

문제 설명

1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 
영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.

1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말합니다.
마지막 사람이 단어를 말한 다음에는 다시 1번부터 시작합니다.
앞사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 합니다.
이전에 등장했던 단어는 사용할 수 없습니다.

사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.

---

제한 사항

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

---

풀이

 

로직

끝말잇기가 성립 하는지 확인

> 중복을 찾을 때 in 말로 try - except, dict로 처리

> cnt 변수를 1씩 증가시키고 divmod를 사용해서 몇번째 사람이 몇번째 말하는 것인지 확인

 

초기 코드

# 초기 코드
def solution(n, words):
    word_dict = {}
    cnt = 0
    last = words[0][0]
    for word in words:
        if last != word[0]:
            turn, people = divmod(cnt,n)
            return [people+1, turn+1]
        try:
            word_dict[word]
            turn, people = divmod(cnt,n)
            return [people+1, turn+1]
        except:
            word_dict[word] = None
            cnt += 1
        last = word[-1]
    return [0,0]
# 0.04ms

 

try-except를 쓴 이유는 예전에 풀었던 코테에서 경험상 try-except가 빠른 경우가 많았기 때문인데

과연 진짜로 그럴까 하는 생각이 들었다.

 

1차 수정 : try-except > if-else

# 1차 수정 : try-except > if-else
def solution(n, words):
    word_dict = {}
    cnt = 0
    last = words[0][0]
    for word in words:
        if last != word[0]:
            turn, people = divmod(cnt,n)
            return [people+1, turn+1]
        if word_dict.get(word, 0):
            turn, people = divmod(cnt,n)
            return [people+1, turn+1]
        else:
            word_dict[word] = 1
            cnt += 1
        last = word[-1]
    return [0,0]
# 0.03ms

 

큰 차이는 없지만 살짝 빨라졌다. 그냥 상황에 따라서 다른 것 같음.

추가로 더 수정할만한 것들이 있나 보면,

• 변수명을 더 명확히
• 중복되는 코드 최소화

 

2차 수정 : 가독성 향상

# 2차 수정 : 가독성 향상
def solution(n, words):
    word_dict = {}
    cnt = 0
    last_spelling = words[0][0]
    for word in words:
        first_spelling = word[0]
        if last_spelling != first_spelling or word_dict.get(word, False):
            turn, people = divmod(cnt,n)
            return [people+1, turn+1]
        word_dict[word] = True
        cnt += 1
        last_spelling = word[-1]
    return [0,0]
# 0.03 ms

 

조건을 합쳐서 코드의 중복을 줄이고, 불필요한 else문을 지웠다.

시간상으론 차이가 없지만, testcase의 공간복잡도가 전반적으로 조금 줄었다.