[Python] 프로그래머스 - 스킬트리(Level2)

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

프로그래머스

내 풀이

def solution(skill, skill_trees):
    answer = 0
    dic = dict()

    for i in range(len(skill)):
        dic[skill[i]] = i

    for i in range(len(skill_trees)):
        check = [-1]
        flag = True
        for j in range(len(skill_trees[i])):
            # 스킬순서에 있는거면
            if skill_trees[i][j] in dic:
                if check[len(check)-1] + 1 != dic[skill_trees[i][j]]:
                    flag = False
                    break
                check.append(dic[skill_trees[i][j]])

        if flag: answer += 1

    return answer

일단 딕셔너리를 한개 만들어서

 

문자열 skill을 인덱스 하나하나 돌면서 딕셔너리에 넣어준다.

 

예를 들어, "CBD"인 skill을 넣어주면 딕셔너리는 {'C': 0, 'B': 1, 'D': 2} 이렇게 만들어진다.

 

 

이제 이중포문을 돌면서

 

스킬순서가 맞는지 확인을 하는데

 

여기서 사용하는 배열이 check라는 배열이다.

 

check는 각각의 skill_trees를 돌면서 하나의 skill_trees 안의 스킬이 skill에 있는 값인지 확인하고 있으면 딕셔너리의 value값을 넣어준다.

 

넣기 전에, check의 맨 마지막 원소에 +1을 한 값이 value값이랑 같지 않으면 

 

순서가 맞지 않는 것이기 때문에 break를 해준다.

 

 

다른 사람 풀이

def solution(skill, skill_trees):
    answer = 0

    for t in skill_trees:
        skill_list = list(skill)
        print(skill_list)

        for s in t:
            if s in skill and s != skill_list.pop(0):
                break
        else:
            answer += 1

    return answer

생각해보니까 in 연산자를 문자열에도 쓸 수 있었다..!

 

그리고 문자열 자료형을 리스트로 바로 변환이 가능하다는 것...!!

 

마지막 대박사건은

 

파이썬이 for-else문이 가능하다는 것이다.....❗❗❗

 

만약에 for문에서 break가 걸렸으면 걍 넘어가는거고

 

for문 끝까지 break가 걸리지 않았으면 else문으로 가는거임!!!!!

 

암튼, 다른 사람의 풀이를 보고 또 많은 것을 배워갑니다,,,