URL : https://programmers.co.kr/learn/courses/30/lessons/49993
문제 설명
- 선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.
- 예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.
- 위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.
- 선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.
제한 조건
- 스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
- 스킬 순서와 스킬트리는 문자열로 표기합니다.
- 예를 들어, C → B → D 라면 CBD로 표기합니다
- 선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
- skill_trees는 길이 1 이상 20 이하인 배열입니다.
- skill_trees의 원소는 스킬을 나타내는 문자열입니다.
- skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다
문제 풀이
- 순서에 상관없이 배울수 있는 스킬은 어디에 있어도 상관이 없고, 선행스킬이 중요하기 때문에 아무때나 배우는 스킬은 삭제함
- 배열 skill_trees에서 선행 스킬 순서가 있는 스킬만 따로 ls에 append 해주어 skill과 일치하는지 확인하는 코드로 작성
- 선행스킬이 모두 끝까지 있는 상황은 아니기 떄문에 (AECB)의 경우 만일 append한 ls가 주어진 skill보다 크기가 작으면 작은 만큼만 비교함
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
def solution(skill, skill_trees):
answer = 0
for skill_tree in skill_trees:
ls = []
for i in skill_tree:
if i in skill:
ls.append(i)
ls = ''.join(ls)
if len(ls) != len(skill):
if skill[:len(ls)] in ls:
answer += 1
else:
if skill in ls:
answer += 1
return answer
1
2
3
4
skill = 'CBD'
skill_trees = ["BACDE", "CBADF", "AECB", "BDA"]
solution(skill, skill_trees)
1
2