[Programmers] 스킬트리

문제 설명

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

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

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

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

제한 조건

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

소스 코드

class Solution {
    public int solution(String skill, String[] skill_trees) {
        int answer = skill_trees.length;
        String[] s = skill.split("");
        // 스킬 순서 각각의 위치를 비교
        for(int i=0;i<skill_trees.length;i++){
            int idx = skill_trees[i].indexOf(s[0]);
            boolean chk = true;
            for(int j=1;j<s.length;j++){
                int tmp = skill_trees[i].indexOf(s[j]);
                // 해당 순서의 스킬이 존재하며, 이전 스킬 인덱스가 더 큰 경우 false
                // 해당 순서의 스킬 인덱스가 존재하는데 이전 스킬 인덱스는 존재하지 않는다면 false
                if((idx > tmp && tmp != -1) || (idx == -1 && tmp != -1)) {
                    chk = false;
                }
                idx = tmp;
            }
            if(chk == false) answer -= 1;  // false면 answer 감소
        }
        return answer;
    }
}

풀이

입출력 예
skill : "CBD"
skill_trees : ["BACDE", "CBADF", "AECB", "BDA"]

skill을 하나씩 나눠 배열에 담고 skill_trees 각각과 비교한다.
indexOf 함수를 이용해 각 스킬의 인덱스를 구한다.
이 때, 아래 두 경우에 해당하면 불가능한 스킬트리이다.

1. 해당 순서의 스킬이 존재하며, 이전 스킬 인덱스가 더 큰 경우 false
   ex) BACDE의 경우 C의 인덱스는 2, B의 인덱스는 0이므로 불가능하다.
2. 해당 순서의 스킬 인덱스가 존재하는데 이전 스킬 인덱스는 존재하지 않는다면 false
   ex) BDA의 경우 C가 존재하지 않고 B는 존재하므로 불가능하다.