[프로그래머스 LV 2] 영어 끝말잇기 파이썬/python

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

코딩테스트 연습 - 영어 끝말잇기

 

 

그냥 주어진 방법대로 구현함...

진짜 주먹구구식으로 구현만 한거라 크게 볼건 없을듯...?

 

 

def solution(n, words):
    answer = [0, 0]
    word_map = {}  # 사용했던적 있는 단어인지 판단용
    temp = words[0][0]
    round_num = 1  # 몇 번쨰 라운드인지
 
    while 1:
        for i in range(n):
            if words[0] not in word_map:
                if temp == words[0][0]:
                    temp = words[0][-1]
                    word_map[words.pop(0)] = 1
                else:
                    result = [i + 1, round_num]
                    return result
            else:
                result = [i + 1, round_num]
                return result
        round_num += 1
        if len(words) == 0:
            break
 
    return answer