코테 준비/Stack, Queue

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

imsmile2000 2024. 1. 30. 14:34

문제 이해하는데 시간이 좀 걸렸다..

def solution(skill, skill_trees):
    cnt=0
    for i in range(len(skill_trees)):
        tree=""
        for j in skill_trees[i]:
            if j in skill:
                tree+=j #skill에 있는 문자만 뽑기       
        flag=True
        if tree=="": # 아예 skill에 없는 문자만 있으면 count
            flag=True
        else:
            if tree[0]!=skill[0]: #가장 첫 알파벳이 선행되어야함
                flag=False 
            if tree not in skill: # 순서대로 스킬을 쌓아야함
                flag=False
        if flag==True:
            cnt+=1
    return cnt