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

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

inistory 2021. 5. 18. 05:54

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

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

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]

programmers.co.kr

def solution(n, words):
    current_word = [words[0]]
    pre_word = words[0]
    for i in range(1,len(words)): 
        #끝말잇기 규칙위반
        if pre_word[-1] != words[i][0]:
            return [(i % n)+1, (i//n)+1]
        #이미 등장한 단어
        elif words[i] in current_word:
            return [(i % n)+1, (i//n)+1]
        else:
            current_word.append(words[i])
            pre_word = words[i]
            
    return [0,0]