https://programmers.co.kr/learn/courses/30/lessons/49993

문제 설명

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

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

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

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

제한 조건

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

풀이

def solution(skill, skill_trees):
    answer = 0
    d = dict()
    for idx, one in enumerate(skill):
        d[one] = idx
    for s in skill_trees:
        arr = []
        for c in s:
            if c in d:
                arr.append(d[c])
        isOk = True
        for i in range(len(arr)):
            if i != arr[i]:
                isOk = False
        if isOk:
            answer += 1
    return answer

• 딕셔너리 d에 키는 skill의 알파벳, value는 인덱스로 저장함 ex) "CBD" -> { C:0, B:1, D:2 }
• s를 한글자씩 쪼개서 만약 d에 한글자씩 쪼갠게 있으면 arr에 그 데이터값을 순서대로 집어넣음
• 결국 arr엔 [0,2,1], [1,2], [0,1] 이런식으로 숫자만 담기게 될꺼임
• 여기서 스킬트리가 성립하려면 [0], [0,1], [0,1,2] 이런식으로 순서대로 와야함
• 즉, arr의 길이가 2이면 반드시 [0,1]만 성립함
• 따라서 arr의 길이를 뽑아서 range로 그 인덱스를 i에 넣어준뒤
• i 랑 arr[i]랑 같은지 비교해주면 됨

다른사람 풀이

def solution(skill,skill_tree):
    answer=0
    for i in skill_tree:
        skillist=''
        for z in i:
            if z in skill:
                skillist+=z
        if skillist==skill[0:len(skillist)]:
            answer+=1
    return answer

• 여기서는 딕셔너리 이딴거 필요없이 skillist라는 빈문자열을 만들고
• 딕셔너리에서 확인하는것이 아닌 skill에 한글자씩 쪼갠게 있나 확인한다음
• 있으면 skillist에 추가해주고
• 마찬가지로 skillist의 길이에 따라 답이 정해져있기 때문에 ex) 2 -> CB, 1 -> C
• 길이만큼 skill을 짤라서 비교해줌

이렇게 간단하게 할 수 있는걸 뭐하러 딕셔너리까지 써가며 복잡하게 풀었는지.. 현타