[프로그래머스][LV.2] 영어 끝말잇기 | python3

 

문제 링크: 영어 끝말잇기

 

문제 설명

문제 설명

1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.

1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말합니다.
마지막 사람이 단어를 말한 다음에는 다시 1번부터 시작합니다.
앞사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 합니다.
이전에 등장했던 단어는 사용할 수 없습니다.
한 글자인 단어는 인정되지 않습니다.

다음은 3명이 끝말잇기를 하는 상황을 나타냅니다.

tank → kick → know → wheel → land → dream → mother → robot → tank
위 끝말잇기는 다음과 같이 진행됩니다.

1번 사람이 자신의 첫 번째 차례에 tank를 말합니다.
2번 사람이 자신의 첫 번째 차례에 kick을 말합니다.
3번 사람이 자신의 첫 번째 차례에 know를 말합니다.
1번 사람이 자신의 두 번째 차례에 wheel을 말합니다.
(계속 진행)

끝말잇기를 계속 진행해 나가다 보면, 3번 사람이 자신의 세 번째 차례에 말한 tank 라는 단어는 이전에 등장했던 단어이므로 탈락하게 됩니다.
사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.

제한사항
- 끝말잇기에 참여하는 사람의 수 n은 2 이상 10 이하의 자연수입니다.
- words는 끝말잇기에 사용한 단어들이 순서대로 들어있는 배열이며, 길이는 n 이상 100 이하입니다.
- 단어의 길이는 2 이상 50 이하입니다.
- 모든 단어는 알파벳 소문자로만 이루어져 있습니다.
- 끝말잇기에 사용되는 단어의 뜻(의미)은 신경 쓰지 않으셔도 됩니다.
- 정답은 [ 번호, 차례 ] 형태로 return 해주세요.
- 만약 주어진 단어들로 탈락자가 생기지 않는다면, [0, 0]을 return 해주세요.

 

---

문제 풀이

같은 단어를 말하거나 전단어의 끝 철자로 시작하지 않으면 끝난다.

같은 단어가 나올때만 고려한 줄 알고 풀다가 급하게 뒤의 조건도 추가해줬다.
그래서 그런가 중복되는 코드도 있고 깔끔하지 않은 느낌. 분명 깔끔하게 작성할 수 있을 것 같은데 조건을 어떻게 줘야 할 지 모르겠다.

나는 일단 사용한 단어의 조건을 먼저 주고 그 다음에 철자의 조건을 줘서 이중 삼중 if문이 완성되었다.

 

내코드

def solution(n, words):
    answer = []
    use_word=[]
    chk=1
    firstend=''
    for word in words:
        if word not in use_word:
            use_word.append(word)
            if not firstend:
                firstend=word[-1]
            else:
                if firstend!=word[0]:
                    if chk%n==0:
                        return [n,chk//n]
                    return [chk%n, chk//n+1] 
                firstend=word[-1]
            chk+=1
        else:
            if chk%n==0:
                return [n,chk//n]
            return [chk%n, chk//n+1]

    return [0,0]

 

 

---

다른사람풀이

def solution(n, words):
    used_words = []
    number, order = 0,0

    used_words.append(words[0])
    last_word = words[0][-1]
    for i in range(1,len(words)):
        if words[i] not in used_words and words[i][0] == last_word:
            used_words.append(words[i])
            last_word = words[i][-1]
        else:
            number = (i%n)+1
            order = (i//n)+1
            break
    
    return [number, order]

내가 조건문을 깔끔하게 정리하지 못한 이유가 맨처음 나오는 단어의 첫글자는 비교할 필요가 없어서 따로 구분을 해줘야 해서 였는데 아예 리스트의 첫단어만 빼서 변수에 넣고 그 다음부터 for문을 돌려서 값을 구해도 되겠구나.

그리고 for문을 인덱스로 돌아서 나처럼 chk라는 변수를 사용하지 않아도 됐음. 0부터 시작하기 나처럼 1부터 시작해서 나눈 값이 0이 될 때를 생각해주지 않아도 된다.

여기서 for문을 enumerate를 사용해서 활용하면 코드가 더 깔끔해질 것 같다.

 

def solution(n, words):
    
    use_word=[]
    use_word.append(words[0])
    last_chr=words[0][-1]
    for idx, word in enumerate(words[1:]):
        if word not in use_word and last_chr==word[0]:
            use_word.append(word)
            last_chr=word[-1]
        else:
            return [(idx+1)%n +1, (idx+1)//n+1]

    return [0,0]

 

이렇게 풀어봤다.

enumerate는 리스트를 저렇게 주면 idx가1부터 시작할 줄 알았는데 아니였다. 리스트 인덱스를 그대로 가져오는줄 알았는데 아니였다. 들어온 값을 다시 새로운 변수로 봐서 0부터 다시 시작하는 듯.

그래서 결과값을 구할때 idx값에 1을 더해주고 풀어야 했다.

 

def solution(n, words):
    answer = []
    turn = 0
    wordList = [words[0]]
    # [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
    for idx in range(1, len(words)):
        if words[idx-1][-1] != words[idx][0]:
            turn = idx
            break
        if words[idx] in wordList:
            turn = idx
            break
        wordList.append(words[idx])
    answer = [turn%n +1, turn//n + 1]
    if turn == 0:
        answer = [0, 0]
    return answer

끝말잇기 틀릴때만 조건을 주면 이중if문을 안써도 된다.

 

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

조건을 끝말잇기 틀릴때만 주고 리스트를 새로 만들지 않고 기존의 words 리스트를 활용하면더 간결하게 가능.