[프로그래머스 파이썬] 영어 끝말잇기

일단 해볼게·2023년 2월 20일
0

프로그래머스

목록 보기
41/106

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

def solution(n, words):
    check = []
    answer = []
    before_word = words[0]
    check.append(before_word)
    
    for i in range(1, len(words)):
        # 전 단어랑 이어지는지, 같은 단어를 말했는지
        if words[i][0] != before_word[-1] or words[i] in check:
            answer.append((i % n) + 1)
            answer.append(i // n + 1)
            break
        else:
            before_word = words[i]
            check.append(before_word)
    
    if len(answer) == 0:
        return [0, 0]
    else:
        return answer
  1. 이전 단어랑 이어지는지
    현재 단어의 첫 문자열(words[i][0])과 이전 단어의 마지막 문자열(before_word[-1])이 같은지 검사
  2. 같은 단어를 말했는지 체크하면 된다.
    check 리스트에 현재 단어가 있는지 검사
profile
시도하고 More Do하는 백엔드 개발자입니다.

0개의 댓글