[Programmers/Python] 영어 끝말잇기

출처

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

소스

def solution(n, words):
    answer = []
    
    flag = 0
    a = 0
    b = 0
    
    for i in range (1, len(words)):
        p1 = words[i-1]
        p2 = words[i]
        
        for j in range(0, i):
            if (words[j] == words[i]):
                flag = 1
                break
                
        len_p1 = len(p1)
        
        if (p1[len_p1 -1] != p2[0]):
            flag = 1
            
        if flag :
            a = (i % n) + 1
            b = int(i / n) + 1
            answer.append(a)
            answer.append(b)
            return answer
        
    answer.append(0)
    answer.append(0)
    return answer