(Python/LV2) 스킬트리

문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/49993

프로그래머스

풀이:

def solution(skill, skill_trees):
    pre_skill = skill[:-1]
    count = 0
    for i in skill_trees:
        if pre_skill in i :
            count += 1
    return count
    
너무 쉽게 생각했다 당연히 안된다

 

정답코드:

def solution(skill, skill_trees):
    answer = 0
    for tree in skill_trees:
        s = ''				# 초기화
        for i in tree:
            if i in skill:  # 선행스킬에 포함된다면
                s += i
        if skill[:len(s)] == s:  # skill의 앞부터 s의 길이만큼  s와 똑같다면
            print(skill[:len(s)] == s)
            answer += 1  
    return answer

 

앞에서 부터 하나씩 확인하여 맞으면 s 에넣어줌

 

마지막에 s와 s 길이만큼의 스킬트리와 맞는지 확인