프로그래머스. 영어 끝말잇기 (Level-2) - 파이썬

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

프로그래머스

 

import math
def solution(n, words):
    answer = [0,0]
    stack = []
  
    for i in range(len(words)):
        if not len(stack): #스택이 비어있으면
            stack.append(words[i])
            continue
        if (stack[-1][-1] !=  words[i][0]) or words[i] in stack : #전 단어의 마지막 글자와 다음 단어의 첫 글자가 다르면 또는 이미 말한 단어를 말했을 경우
            return [i%n+1,math.ceil((i+1)/n)]
        else : #같으면 
            stack.append(words[i])
    #문제없이 성공했을 경우는 0,0 리턴
    return [0,0]

문제를 보자마자 스택을 만들어서 스택의 마지막 원소의 마지막 글자와 단어의 첫 번째 글자를 비교해주면 되겠다..라고 생각해서 풀었다. 정답이긴 하나 

def solution(n, words):
    for i in range(1, len(words)):
        if words[i][0] != words[i-1][-1]  or words[i] in words[:i] :
            return [(i%n)+1, (i//n)+1]
    else:
        return [0,0]

이게 다른 사람들 풀이인데... 나 왜 굳이 스택을 만들어줬지? 그냥 word자체에서 idx-1과 idx를 비교해주면 되네 

코드도 진짜 가독성 좋고 훨씬 간단하다