728x90
반응형
출처 : 프로그래머스
선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.
예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.
위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.
선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.
제한 조건
- 스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
- 스킬 순서와 스킬트리는 문자열로 표기합니다.
- 예를 들어, C → B → D 라면 "CBD"로 표기합니다
- 선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
- skill_trees는 길이 1 이상 20 이하인 배열입니다.
- skill_trees의 원소는 스킬을 나타내는 문자열입니다.
- skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.
입출력 예
skillskill_treesreturn
"CBD" | ["BACDE", "CBADF", "AECB", "BDA"] | 2 |
입출력 예 설명
- "BACDE": B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트립니다.
- "CBADF": 가능한 스킬트리입니다.
- "AECB": 가능한 스킬트리입니다.
- "BDA": B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.
- 스킬 트리: 유저가 스킬을 배울 순서 ↩
코딩테스트 연습 - 스킬트리
programmers.co.kr
- index를 찾는다. 해당 index가 있을 경우만 return 한다.
- index가 있는 값만 result을 추가한다.
- result의 길이가 0일 경우에는 answer의 값을 추가한다.
- match_skill 길이만큼 list를 생성한다. 예를 들어 아래 케이스의 경우 : [0, 1, 2, 3]
- 정렬된 result과 result길이 만큼 짤린 match_skill 의 값이 같을 경우 예를 들어 result=[0,1] result과 result길이 만큼 짤린 match_skill 의 값이 같을 경우 result=[1,0]을 방지하기 위해서
def solution(skill, skill_trees):
answer = 0
match_skill = "".join([str(i) for i in range(len(skill))])
for skill_tree in skill_trees:
result= ""
for i in skill_tree:
skill_index = skill.find(i)
if (skill_index) != -1:
result += str(skill_index)
if result == match_skill[:len(result)]:
answer+= 1
return answer
다른 사람 풀이를 보니깐 매우 간단하게 하였다. 좀 너무 어렵게 접근한 것 같다.
그래서 코드를 좀 수정하였다.
반응형
'문제 > 프로그래머스' 카테고리의 다른 글
n^2 배열 자르기 (0) | 2022.01.15 |
---|---|
쿼드압축 후 개수 세기 (0) | 2022.01.06 |
방문 길이 (0) | 2022.01.03 |
[3차] 방금그곡 (0) | 2021.12.30 |
가장 큰 정사각형 찾기 (0) | 2021.12.24 |