[오늘의 코테 연습장] - 스킬트리 ( 자바 )

문제 설명

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

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

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

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

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

문제 풀이 과정

처음에는 skill에 순서 idx를 부여해서 숫자가 순서대로 정렬되어있지 않으면 pass 하고,
숫자가 순서대로 정렬되어있으면 count하게 하자라는 생각이 들었다.
그럼 SKILL에 저장되어있는 알파벳들과 idx를 같이 부여하자! 라는 생각이 들어 HashMap을 이용하여 구현하였다.
이후 스킬 트리 가능 여부 변수를 선언하여
스킬 트리를 순회하여 스킬Map에 해당 스킬이 존재하지 않는 경우에는 pass
스킬Map에 스킬이 존재하는 경우에만 스킬 트리 count 증가를 수행한다.
for문을 돌며 현재 스킬의 인덱스를 get해와서
현재 스킬 - 1 = 이전 스킬
즉 이전 스킬이 아니라면 스킬 트리 가능 여부 변수를 fasle 하고 break 한다.
만약에 이전 스킬이 맞다면 이전에 배운 스킬 인덱스를 현재 스킬의 인덱스로 바꾸며 for문을 진행한다.
만약 isEnable이 true라면 count를 증가시켜 count를 반환한다.

구현코드

import java.util.HashMap;

class Solution {
    public int solution(String skill, String[] skill_trees) {
        // 스킬에 idx 부여
        // 스킬과  그에 해당하는 인덱스를 Map에 저장
        HashMap<Character,Integer> skillMap  = new HashMap<>();
        for(int i = 0; i < skill.length(); i++){  
            char s = skill.charAt(i);
            skillMap.put(s, i);
        }
        
        int count  = 0;
        for(String s : skill_trees){
            // 스킬트리 가능한지 여부
            boolean isEnable = true;
            // 이전 인덱스 
            int preidx = -1;
            for(int i = 0 ; i < s.length(); i++){
                char c = s.charAt(i);
                // 스킬맵에 스킬이 존재하는 경우에만 처리
                if(!skillMap.containsKey(c)){
                    continue;
                }
                //스킬의 인덱스를 get
                int curidx = skillMap.get(c);
                // 이전에 배운 스킬이 아니라 나중에 배워야 하는 스킬이라면 
                // 불가능한 스킬이다.
                if(preidx != curidx - 1){
                    isEnable = false;
                    break;
                }
                // 이전에 배운 스킬의 인덱스를 현재 스킬의 인덱스로 바꿈
                preidx = curidx;
            }
            if(isEnable){
                count ++;
            }
        }
        return count;
    }
}