본문 바로가기

CodingTest/Level2

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

코딩테스트 연습

  • programmers.co.kr
  • 서머코딩/윈터코딩 (2018)
  • 스킬트리 문제

 


 

def check(x,skill):
    tmp_dict = {}
    for i, v in enumerate(skill):
        if v in x:
            tmp_dict[i] = x.index(v)

    aa = sorted(tmp_dict.items(), key = lambda x: x[1])
    tmp = -1

    f = 0
    for x in aa:
        if f == 0 and x[0] != 0:
            return 0
        else:
            if tmp > x[0]:
                return 0
            if x[0]-tmp != 1:
                return 0
            tmp = x[0]
            f = 1

    return 1

def solution(skill, skill_trees):
    cnt = 0
    for x in skill_trees:
        if check(x,skill):
            cnt +=1

    return cnt
반응형