이숭간 공부기록

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

알고리즘/프로그래머스

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

이숭간 2021. 7. 15. 21:06
728x90

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

문제유형 : 문자열 

 

문제풀이

주어진 조건을 잘 맞게 구현해야한다.

문제에서 말로 설명되어있는것은 꼭 체크해서 맞게 풀어줘야한다!!!

 

정답코드 :

def solution(n, words):
    answer = []
    rounds = []
    before = words[0][0]
    prev = []
    
    # 주어진 배열을 사람숫자로 나누어 2차원배열로 
    # 2차원배열의 행이 몇번째인지를 나타낸다.
    for i in range(0,len(words)+1, n):
        temp = words[i:i+n]
        rounds.append(temp)
                      
    print(rounds)
                      
    for round, words in enumerate(rounds):
        for person, word in enumerate(words):
            if word not in prev and len(word)>1 and word[0]==before:
                prev.append(word)
                before = word[-1]
            else:
                return [person+1, round+1]
    
    return [0,0]
    return answer