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

Python3, 구현

문제

제한 사항

입출력 예시

나의 풀이 (시도)

• 끝말잇기를 실패한 조건문에서 바로 리턴해주지 않아 테스트 케이스에서 오류가 발생했다
  -> 바로 리턴을 해주도록 수정하고 그 외 성공한 경우는 따로 0, 0을 리턴했다

주요 포인트

끝 글자와 시작하는 글자를 비교하려면
words[i][0] != words[i-1][-1]

이미 나온 단어인지 판단하려면
words[i] in words[:i]

i전까지 단어 중에서 i의 인덱스 값과 같은 단어가 있는지 판단

최종 코드

def solution(n, words):    
    for i in range(1, len(words)):
        # 만약 끝말잇기가 끝 글자로 시작하지 않으면 or 이미 나온 단어면
        if words[i][0] != words[i-1][-1] or words[i] in words[:i]:
            return [(i % n) + 1, (i // n) + 1]
            
    return [0, 0]