문제 설명
선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.
예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.
위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.
선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.
제한 조건
스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
스킬 순서와 스킬트리는 문자열로 표기합니다.
예를 들어, C → B → D 라면 "CBD"로 표기합니다
선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
skill_trees는 길이 1 이상 20 이하인 배열입니다.
skill_trees의 원소는 스킬을 나타내는 문자열입니다.
skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.
입출력 예
| skill | skill_trees | return |
|---|---|---|
| "CBD" | ["BACDE", "CBADF", "AECB", "BDA"] | 2 |
코드
from collections import deque
def solution(skill, skill_trees):
count = 0
for skills in skill_trees:
skill_list = deque(skill)
for item in skills:
if item not in skill_list:
continue
if item == skill_list[0]:
skill_list.popleft()
else:
break
else:
count += 1
return count배운 점
일단 이 문제의 두번째 풀이인데 저번보다 더 정돈된 코드로 풀이한 것 같아서 뿌듯!
처음에는 조금 헤매다가 skill trees의 원소가 중복되지 않는다는 것을 보고 힌트를 얻었다.
skill_trees에 주어진 스킬들이 skill 배열안의 원소와 순서가 동일하게 주어져야하기 때문에 skill 배열을 deque로 만들어주고 skill_trees 안의 문자열을 하나씩 돌면서 확인했다.
만약 skill 배열 안에 없는 스킬이라면 그냥 건너뛰기 위해 continue하고, 만약에 skill 배열 안에 있고 skill 배열의 첫번째 원소와 동일하다면 순서가 잘 진행되고 있는 것이기 때문에 skill 배열에서 popleft를 하여 해당 스킬을 제거해주었다.
만약 skill 배열 안에 있지만 첫번째 원소와 다를 경우, 순서가 뒤섞였다는 뜻이기 때문에 break를 해주었다. 또한 for-else문을 사용해주어 for문을 다 돌았을 때 break를 하지 않은 경우에 count += 1을 해주었다.
코드를 쓰기 전에 생각을 충분히 하는게 중요하다는걸 깨닫게 해준 문제..~!
