[프로그래머스 문제] 스킬 트리

문제 설명

선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.

예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.

위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.

선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.

제한 조건
스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
스킬 순서와 스킬트리는 문자열로 표기합니다.
예를 들어, C → B → D 라면 "CBD"로 표기합니다
선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
skill_trees는 길이 1 이상 20 이하인 배열입니다.
skill_trees의 원소는 스킬을 나타내는 문자열입니다.
skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

입력 및 출력

skilㅣ: "CBD"
skill_trees: ["BACDE", "CBADF", "AECB", "BDA"]
return: 2

내가 생각한 풀이 전략

• 큐를 사용해서 풀어야 할 것 같다. (데이터를 선입선출 방식으로 넣고 순차적으로 비교하면 될 것으로 생각.)
• 이중 반복문을 통해 skill의 데이터와 skill_trees의 데이터를 비교, 입출력 예시를 예로 들자면 C와 BACDE를 비교할 때 C가 BACDE안에 있는지와 그 순서 값을 저장하여 다음 B를 비교할 때 B의 순서가 ㅊ의 순서보다 앞에 있으면 다음인 CBADF와 비교하는 방식.

결과

실패(일부 테스트 케이스 통과 x)

skill = "CBD"
skill_trees = ["BACDE", "CBADF", "AECB", "BDA"]

def solution(skill, skill_trees):
    queue = []  # skill값을 받을 큐
    queue_compare = []  # skill_trees 값을 받을 큐
    k = 0
    answer = 0

    for i in range(len(skill)):
        queue.append(skill[i])
    for j in range(len(skill_trees)):

        value = skill_trees[j]
        # print("ttt ", type(skill_trees[j]) )
        for k in range(len(value)):
            queue_compare.append(value[k])  # 값 다 집어넣음.
        one = -1
        count = 0
        check = True
        for q in range(len(queue)):
            for z in range(len(queue_compare)):
                if (queue[q] == queue_compare[z]):
                    count += 1
                    if (z > one):
                        one = z
                    else:
                        one = 9999999
            if(count == 0):
                check = False
        queue_compare.clear()
        if (one != 9999999 and check == True):
            answer += 1

    print(answer)


solution(skill, skill_trees)

다른 사람의 풀이

from collections import deque
skill = "CBD"
skill_trees = ["BACDE", "CBADF", "AECB", "BDA"]

def solution(skill, skill_trees):
    answer = 0
    for tree in skill_trees:
        queue = deque(skill)
        whether = True

        for s in tree:
            if s not in queue:
                continue
            else:
                if s == queue[0]:
                    queue.popleft()
                else:
                    whether = False
                    break
        if whether:
            answer += 1

    return answer

solution(skill,skill_trees)

복기

• 다른 사람의 코드가 훨씬 간결하였다. (내 풀이 접근법은 옳았다.)
• 테스트 케이스 예외가 발생한 경우를 생각해보았을 때, 데이터의 순서는 맞았으나 필수적인 데이터가 모두 들어가 있지 않은 경우를 제외해야 하지만 제외하지 못한 것으로 추측.