[Programmers] 스킬트리 (Python 3)
문제 설명
선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.
예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.
위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.
선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.
제한 조건
- 스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
- 스킬 순서와 스킬트리는 문자열로 표기합니다.
- 예를 들어, C → B → D 라면 CBD로 표기합니다
- 선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
- skill_trees는 길이 1 이상 20 이하인 배열입니다.
- skill_trees의 원소는 스킬을 나타내는 문자열입니다.
- skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.
문제 파악은 끝났다.
첫번째로 내가 생각한 방법은 Queue를 이용한 것이었다.
- 일단 skill 문자열을 Queue 형태로 만들어둔다.
- 그리고 skill_trees의 요소 하나당
- 알파벳 별로 skill에 있는 문자열에 해당하는지 조건문 돌린다
- 큐에서 dequeue 한 값과 같은지 비교하는걸 반복
그래서 초안은
내 풀이
from collections import deque
def solution(skill, skill_trees):
answer = 0
for tree in skill_trees:
q = deque(list(skill))
flag=1
for t in tree:
if t in skill:
if t != q.popleft():
flag=0
break
if flag:
answer+=1
return answer이렇게 해서 통과를 하긴 했다.
근데 다른 사람들의 풀이 중 인상깊은 코드를 발견했다.
다른 사람의 풀이 - 1
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전체적인 로직은 내 풀이와 비슷하다. 근데 else문의 위치를 보고 눈을 의심했다.
python에 for-else라는 구문도 있는지 처음.. 알았다....
for-else에서 else는 for문에서 break로 중간에 빠져나와지지 않았을 때 수행하는 작업이다.
이걸 알았더라면 내 코드에서 flag를 따로 둘 필요가 없었을 것이다.
알게 된 이상 꼭 써먹어보자
다른 사람의 풀이 - 2
def solution(skill,skill_tree):
answer=0
for i in skill_tree:
skillist=''
for z in i:
if z in skill:
skillist+=z
if skillist==skill[0:len(skillist)]:
answer+=1
return answer가장 깔끔하고 내 취향에 맞는 코드다
일단 해당하는 알파벳을 따로 모두 뽑아놓고 원 skill 문자열과 비교한다는걸 생각하지 못했다.
뭐 장단점이 있겠지만.. 지금 나에게 코드를 짜기 전에 좀 더 구조를 신중히 생각하는 연습이 필요한건 맞는것 같다.
공부하자 공부