[프로그래머스] 스킬트리 - python

문제

• 프로그래머스 2018 서머코딩- 스킬트리
• https://programmers.co.kr/learn/courses/30/lessons/49993

풀이

선행 스킬 순서 skill과 유저들이 만든 스킬트리를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 해야한다.

중복이 주어지지 않기 때문에 skill_tree에 값을 하나씩 비교하여 temp에 담고 주어진 skill과 동일한지 비교하여 판단한다.

만약 temp와 skill의 값이 일치하면 가능한 스킬트리로 간주하여 answer+1 해줌.

코드

def solution(skill, skill_trees):
    answer = 0
    for i in skill_trees:
        temp = list()
        check = True;

        for j in range(len(i)):
            if i[j] in skill:
                temp.append(i[j])
        for k in range(len(temp)):
            if temp[k] != skill[k]:
                check = False
                break
        if check == True:
            answer +=1
    return answer