👨🏻‍💻알고리즘/프로그래머스

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

waveofmymind 2023. 3. 10. 09:08

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/12981

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

코드

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

 

풀이

  • 시퀀스마다 현재 단어의 맨 앞자리와 이전의 단어의 끝자리를 비교해야 하므로 for문의 인덱스를 1부터 설정했다.
  • 탈출 조건은 단어의 앞자리와 이전의 단어의 끝자리가 다르면서, 현재 단어가 이전에 사용된 적이 있을때이다.
  • 탈출시 리턴으로는 탈락한 사람의 순서를 구하기 위해 i%n + 1, 몇번이나 반복됬는지를 구하기 위해 i//n + 1을 했다.