[PRGMS] 영어 끝말잇기

BbickBbick_Develop·2022년 9월 24일
0

PRGMS lv2

목록 보기
2/3
post-thumbnail

정답코드

def solution(n, words):
    answer = [0, 0]
    gone = [[] for _ in range(n)]
    word = []
    for i in range(len(words)):
        # 이미 말한 단어를 말한다면
        if words[i] in word:
            answer[1] = len(gone[i%n])+1
            answer[0] = i%n+1
            break
        # 끝 글자와 맞지 않는 글자를 말한다면
        elif len(word)>0 and words[i][0] != word[-1][-1]:
            answer[1] = len(gone[i%n])+1
            answer[0] = i%n+1
            break
        else:
            gone[i%n].append(words[i])
            word.append(words[i])
    return answer

몇 가지 고려 사항이 있는 걸 감안하면 쉬운 문제(레벨 1수준?)

profile
삑삑도요가 되자

0개의 댓글