파이썬 알고리즘-114 (프로그래머스) 영어 끝말잇기

jiffydev·2021년 2월 24일
0

Algorithm

목록 보기
121/134

코드

def solution(n, words):
    answer = []
    res=[words[0]]
    p=2
    for i in range(1,len(words)):
        if words[i] in res:
            cnt=(i+1)//n if (i+1)%n==0 else ((i+1)//n)+1
            answer=[p,cnt]
            break
        else:
            res.append(words[i][0])

        if words[i-1][-1]==words[i][0]:
            if p<n:
                p+=1
            else:
                p=1
        else:
            cnt=(i+1)//n if (i+1)%n==0 else ((i+1)//n)+1
            answer=[p,cnt]
            break
    else:
        answer=[0,0]

    return answer
# 테스트케이스 일부 통과 못한 코드

다른 사람의 풀이

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, 1+(i//n)]
    else:
        return [0,0]
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글