[ Programmers / CodingTest / Python ] 스킬트리

황승환·2022년 1월 15일
0

Python

목록 보기
95/498

문제 설명

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

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

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

선행 스킬 순서 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

입출력 예 설명

  • "BACDE": B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트립니다.
  • "CBADF": 가능한 스킬트리입니다.
  • "AECB": 가능한 스킬트리입니다.
  • "BDA": B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.
  1. 스킬 트리: 유저가 스킬을 배울 순서 ↩

접근 방법

skill_trees를 순회하며 해당 스킬이 skill에서 가리키는 위치의 스킬과 동일한 경우, 해당 스킬이 skill에 없는 경우, 해당 스킬이 skill에 있지만 skill에서 가리키는 위치의 스킬과 다른 경우 이렇게 3가지로 나누어서 마지막 경우에 chk를 False로 변환하여 카운팅하지 않도록 구현하였다.

  • answer를 0으로 정의한다.
  • skill_trees의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    -> skill의 포인터로 사용할 level을 0으로 정의한다.
    -> skill_trees[i]의 포인터로 사용할 idx를 0으로 정의한다.
    -> 가능한 스킬트리인지 확인하기 위한 변수 chk를 True로 정의한다.
    -> idx가 skill_trees의 길이보다 작고, level이 skill의 길이보다 작을 동안 반복하는 while문을 돌린다.
    --> 만약 skill_trees[i][idx]가 skill[level]과 같다면 level을 1 증가시키고, idx를 1 증가시킨다.
    --> 만약 skill_trees[i][idx]가 skill에 없을 경우 idx만 1 증가시킨다.
    --> 이 외의 경우에는 chk를 False로 갱신하고 while문을 탈출한다.
    -> 만약 chk가 True라면 answer를 1 증가시킨다.
  • answer를 반환한다.

solution.py

def solution(skill, skill_trees):
    answer = 0
    for i in range(len(skill_trees)):
        level=0
        idx=0
        chk=True
        while idx<len(skill_trees[i]) and level<len(skill):
            if skill_trees[i][idx]==skill[level]:
                level+=1
                idx+=1
            elif skill_trees[i][idx] not in skill:
                idx+=1
            else:
                chk=False
                break
        if chk==True:
            answer+=1
    return answer

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글