programming

프로그래머스 스킬트리 파이썬 풀이 및 해설


by Kitle · 2020. 07. 20.



출처 프로그래머스 : https://programmers.co.kr/learn/courses/30/lessons/49993 (Summer/Winter Coding(~2018)

소스 깃허브에서 보기 : https://github.com/kitlexyz/algorithm/blob/master/programmers/courses_30_lessons_49993.py

난이도 : Level2 (max5)

문제 설명

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


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


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


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


제한 조건

스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.

스킬 순서와 스킬트리는 문자열로 표기합니다.

예를 들어, C → B → D 라면 CBD로 표기합니다

선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.

skill_trees는 길이 1 이상 20 이하인 배열입니다.

skill_trees의 원소는 스킬을 나타내는 문자열입니다.

skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

입출력 예

skill skill_trees return

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

입출력 예 설명

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

CBADF: 가능한 스킬트리입니다.

AECB: 가능한 스킬트리입니다.

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

[풀이]

def solution(skill, skill_trees):
answer = 0
seq = []
for item in skill:
seq.append(item)

for skill_tree in skill_trees:
checker = True
temp_seq = seq.copy()
for letter in skill_tree:
if letter in temp_seq and temp_seq[0] == letter:
temp_seq.remove(letter)
elif letter in temp_seq and temp_seq[0] != letter:
checker = False
break
if checker is True:
answer += 1

return answer

[해설]
결론적으로 스킬 트리가 주어지고, 유저가 해당 순서(sequence)를 맞게 가고 있느냐가 중요합니다. 꼭 100% 매칭할 필요는 없고, 매칭이 틀어진 경우만 False 로 처리하면 되겠습니다.

정상적인 케이스 먼저 보시면 CBD 가 스킬 트리이고 CBADF 순서로 찍는다고 하겠습니다.
사실 중간에 다른 letter 즉 CBD 가 아닌 것은 오든 말든 무관합니다.
따라서 스킬트리의 레터가 아니면 검사할 필요가 없습니다. 이런 don't care 조건은 skip 하도록 합니다.
letter in skill_tree:
스킬트리는 리스트로 하나씩 분리하였고, 사용자의 문자가 분리된 스킬트리의 리스트에 포함되는지를 체크 했습니다.
그다음은 0번 인덱스 부터 순서대로 비교해주면 됩니다.
C -> B -> D 순으로 와야 합니다.
사용자 글자가 C 라면 정상입니다. 이 경우 해당 글자를 제거해 줍니다. 비교를 마쳤으니 다음 0번 순을 B로 만들어주기 위함입니다.
그다음 텍스트가 B가 왔습니다. 또다시 0번 위치와 비교하는데 위에서 C가 제거되었으므로 B와 비교됩니다.
이런식으로 순서대로 됩니다.
만약 다른 텍스트가 온다면 순서가 잘 못 된 것이니 False 처리 해줍니다.
만약 순서는 맞지만 C -> B 하고 D가 남게 되어도 잘못 찍은 것은 아니니 이것도 올바른 스킬트리로 판단해주면 되겠습니다.
리스트에서 remove를 했으므로 다음 사용자 스킬트리를 할때 다시 비교 원본을 초기화 해주기 위해 copy()를 사용했습니다.

약간 좋은 풀이는 아닌것 같네요 :)