스킬트리

프로그래머스 - 스킬트리

문제 설명

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

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

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

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

제한 조건

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

입출력 예

해결 과정

1. 스킬 트리에서 문자를 하나씩 가져와야 한다.
   어찌되었든 skill 문자열을 통해 선행 스킬 조건을 잘 지키고 있는지 확인하기 위해서는 각각의 스킬 트리에서 문자를 순서대로 가져와 비교하는 것이 가장 먼저 떠올랐다.

2. 비교하는 문자는 반드시 skill에 포함되어 있어야 한다.
   스킬 트리를 보면 선행 스킬과 무관한 스킬은 언제든지 배울 수 있기 때문에 문자를 비교할 때 skill 안에 있는 문자인지 검사를 진행한 다음 비교해야 한다.

다른 사람의 풀이

하지만, 이 문제를 해결할 수 없어 다른 사람의 풀이를 통해 정답을 확인했다.

def solution(skill, skill_trees):
    answer = 0

    for skills in skill_trees:
        skill_list = list(skill)

        for s in skills:
            if s in skill:
                if s != skill_list.pop(0):
                    break
        else:
            answer += 1

    return answer

skill을 리스트로 바꿔서 사용하는데, 그 이유는 리스트의 메서드 중 하나인 pop을 사용하기 위해서다. skill에서 한 번 검사한 스킬은 차후 등장하지 않아야 조건을 만족하는 것이기 때문에 다시 검사할 필요가 없다. 그래서 pop을 통해 조건 검사와 동시에 리스트에서 제거하는 것이다.

for ... else 문을 사용해 훨씬 간결하게 구현하기도 했다. for ... else는 반복문이 break에 의해 중단되지 않았을 경우 else 구문을 실행한다. 위 코드의 경우 반복문이 중단되지 않았다는 것은, 해당 스킬 트리는 선행 스킬 순서를 잘 만족했다는 것을 의미하기에 answer 값을 1 증가시킨다.