programmers - 영어끝말잇기

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

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

def solution(n, words):
    temp =[]
    temp.append(words[0])
    ans =[]
    for i in range(1,len(words)):
        if words[i] in temp or words[i-1][-1] != words[i][0]:
            #print(words.index(words[i]))
            if (i+1)%n==0:
                ans.append(n)
            else:
                ans.append((i+1)%n)
            turn = (i+1)/n
            if int(turn) != turn: #이부분 수정
                turn = int(turn) + 1

            ans.append(turn)
            break
        temp.append(words[i])
    if len(ans) == 0:
        ans.append(0)
        ans.append(0)
    

    return ans