프로그래머스: 스킬트리

문제

코딩테스트 연습 - 스킬트리

 

문제 풀이

 스킬을 배워야 하는 순서가 주어지고, 스킬을 배운 목록이 주어질 때 해당 스킬 트리가 스킬을 배우는 순서에 맞게 배웠는지 확인하는 문제이다. 문제를 풀기 위해, 스킬 트리에서 순서에 영향을 미치는 스킬들만 추려낸 후 배워야 하는 순서와 일치하는지 확인하면 문제를 쉽게 할 수 있다.

 

코드

def solution(skill, skill_trees):
    answer = 0

    for skill_tree in skill_trees:
        cur_tree = [s for s in skill_tree if s in skill]
        for idx in range(len(cur_tree)):
            if cur_tree[idx] != skill[idx]:
                break
        else:
            answer += 1
    return answer

 반복문이 중단되지 않고, 모두 순회한다면 가능한 스킬 트리이므로 answer를 1 증가시킨다.