[프로그래머스] Summer/Winter Coding(~2018) - 스킬트리

programmers.co.kr/learn/courses/30/lessons/49993?language=python3

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

 

문제 설명

선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.

예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.

위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.

선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.

제한 조건

• 스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
• 스킬 순서와 스킬트리는 문자열로 표기합니다.
  • 예를 들어, C → B → D 라면 CBD로 표기합니다
• 선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
• skill_trees는 길이 1 이상 20 이하인 배열입니다.
• skill_trees의 원소는 스킬을 나타내는 문자열입니다.
  • skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

입출력 예

skillskill_treesreturn

"CBD"

["BACDE", "CBADF", "AECB", "BDA"]

2

입출력 예 설명

• BACDE: B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트립니다.
• CBADF: 가능한 스킬트리입니다.
• AECB: 가능한 스킬트리입니다.
• BDA: B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.

 

---

 

풀이

skill 문자열을 한글자 씩 뽑아 새로운 리스트의 -1번째에서
문자열에서 추출한 글자가 몇번째 인덱스에 있는지 담는다.
그게 오름차순으로 정렬되어있지 않으면 불가능한 스킬트리
else면 가능한 스킬트리

 

선행관계 조건을 넣어야함
skill문자열이 전부 포함되어 있을 때와 그렇지 않은 경우를 구별?
포함되어 있지 않을때의 조건
1. 그 문자가 맨마지막 값이라면 없든 말든 상관없음
2. 첫 값이라면 없으면 무조건 불가능함
3. 중간값이 없는데 내 뒷값이 있다 불가능

3==> 뒷값을 뽑아서 리스트에 담음 그리고 if in 을 이용하여
있으면 False

for - else ==> for문이 중간에 끊기지않고 수행되면 else문 실행

 

def solution(skill, skill_trees):
    answer = 0
    
    for tree in skill_trees:
        tree = list(tree)
        skill_list = list(skill)
        chk = True
        while tree:
            item = tree.pop(0)
            if item in skill_list:
                s = skill_list.pop(0)
                if s != item:
                    chk = False
                    break
        if chk:
            answer += 1
        
            
    return answer