[프로그래머스/Level2/파이썬3(python3)] 스킬트리

[프로그래머스/Level2/파이썬3(python3)] 스킬트리

문제

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

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

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

선행 스킬 순서 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 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.

풀이

def solution(skill, skill_trees):
    answer = 0
    for tree in skill_trees:
        current_tree_skill = [char for char in tree if char in skill]
        possibility = True
        for i, j in zip(current_tree_skill, skill):
            if i != j:
                possibility = False
                break
        if possibility:
            answer += 1
    return answer

설명

일단 주어진 스킬트리 배열에서 스킬 순서와 관련있는 스킬만 남기는 과정을 진행한 뒤에 관련스킬만 남긴 문자열과(current_tree_skill) 스킬순서(skill)를 zip함수로 동시에 돌리면서 전부 일치하면 스킬트리에 문제가 없는 것이므로 answer에 1을 더하고, 일치하지 않는 순간 불가함을 판단하고 다음 스킬트리를 조사하는 방향으로 하였다. zip함수는 두개의 같은 자료형을 묶어주는 함수이지만 요소의 갯수에 차이가 날 땐 적은 쪽을 따라가므로 주어진 스킬 순서 전부가 들어있지 않더라도 문제 없다. 뭔가 깔끔하지 않아보이는 로직이지만 다른 방법은 잘 떠오르지 않는다. 다른 코드들을 봐도 로직이 크게 다르지 않아 이게 최선인듯 싶다.