[파이썬/python] 프로그래머스 : 영어 끝말잇기 (Lv.2)

https://school.programmers.co.kr/learn/courses/30/lessons/12981?language=python3 

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

for문의 범위를 1번부터 words의 길이만큼 지정하고,

words[i-1][-1]과 words[i][0]비교하여 같지 않거나, words[i]가 words[:i]의 배열 안에 없을 경우, (i의 바로 직전까지의 배열안에 없을 경우)

(i%n) + 1, i//n+1을 반환한다

for문을 다 돌 때까지 조건에 해당하지 않으면 [0,0]을 반환한다.

 

comment