프로그래머스 [Lv. 2] 영어 끝말잇기 {언어 : Python}

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12981?language=python3

프로그래머스

정답 코드

def solution(n, words):
    answer = [0, 0]
    prevWords = dict()
    lenW = len(words)
    
    kdx = 0
    ldx = 0
    lastW = ""
    while kdx < lenW:
        mdx = kdx % n
        currentWord = words[kdx]
        f, l = currentWord[0], currentWord[-1]
        
        if kdx == 0:
            prevWords[currentWord] = 1
            lastW = l
        else:
            if lastW == f:
                if prevWords.get(currentWord, 0):
                    answer[0] = mdx + 1
                    answer[1] = ldx + 1
                    break
                else:
                    lastW = l
                    prevWords[currentWord] = 1
            else:
                answer[0] = mdx + 1
                answer[1] = ldx + 1
                break
        
        if kdx and mdx == n - 1:
            ldx += 1
            
        kdx += 1
    return answer

풀이 방법

1. 사실상 mdx와 ldx는 각각 몫과 나머지이다.
2. 몫과 나머지를 바탕으로 처음과 끝이 이어지지 않는 부분에서 answer를 치환한다.
3. 중간에 이미 말한 단어인지만 체크하면 끝! 

느낀점

• 쉬운 문제를 조금 복잡하게 풀어서 아쉬웠다.