프로그래머스 : 영어 끝말잇기

2021. 10. 21. 05:13문제풀기/프로그래머스

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

 

코딩테스트 연습 - 영어 끝말잇기

3 ["tank", "kick", "know", "wheel", "land", "dream", "mother", "robot", "tank"] [3,3] 5 ["hello", "observe", "effect", "take", "either", "recognize", "encourage", "ensure", "establish", "hang", "gather", "refer", "reference", "estimate", "executive"] [0,0]

programmers.co.kr

반복문을 돌리다가 주어진 조건을 만족하지 않으면 해당이 몇 번째 i인지 a에 저장하고 반복문을 나간다.

이 후, a//n+1을 하면 차례가 나오고, a%n을 이용하면 번호가 나온다.

 

def solution(n, words):
    answer = []
    a = -1
    for i in range(1,len(words)):
        if words[i-1][-1] != words[i][0]:
            a = i
            break
        
        if len(words[i]) == 1:
            a = i
            break
            
        if words[i] in words[:i]:
            a = i
            break
            
    print(a)
    if a == -1:
        answer = [0,0]
    else:
        if (a+1)%n == 0:
            answer.append(n)
        else:
            answer.append((a+1)%n)
            
        answer.append(a//n+1)

    return answer