[Python] 스킬트리

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

풀이

1. 스킬트리 안의 각 스킬들이 선행스킬안에 있는지 파악. 있으면 다른 리스트 하나에 저장.

2. 그 리스트에서 enumerate를 해서 idx, 값 추출.

3. 그 스킬과, 그 스킬의 인덱스에 해당하는 선행스킬 리스트의 스킬과 같지 않으면 불가능한 스킬트리인 것.

 

def solution(skill, skill_trees):
    skill = list(skill)
    flag = []
    
    for i in skill_trees:
        post_skill_trees = []
        for j in list(i):
            if j in skill:
                post_skill_trees.append(j)
        
        for idx, j in enumerate(post_skill_trees):
            if j != skill[idx]:
                flag.append(-1)
                break
    
    return len(skill_trees) + sum(flag)