[프로그래머스] 스킬트리 (level 2)

프로그래머스 스킬트리

스킬트리

문제 설명

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

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

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

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

제한 조건

스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
스킬 순서와 스킬트리는 문자열로 표기합니다.
예를 들어, C → B → D 라면 CBD로 표기합니다
선행 스킬 순서 skill의 길이는 2 이상 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 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.

public class SkillTree {  
public int solution(String skill, String\[\] skill\_trees) {  
int answer = 0;
   int treeLength = skill_trees.length;
    for(int i=0; i<treeLength; i++){
        int skillIndex=0;
        boolean flag = true;

        int treeIdxLength = skill_trees[i].length();
        for(int j=0; j<treeIdxLength; j++){

            int skillLength = skill.length();
            for(int k=skillIndex; k<skillLength; k++){
                if(skill.charAt(k) == skill_trees[i].charAt(j)){
                    if(k!=skillIndex){
                        flag = false;
                    }else{
                        skillIndex++;
                    }

                }
            }
        }

        if(flag == true){
            answer ++;
        }
    }



    return answer;
}

/*public static void main(String[] args) {
    SkillTree s = new SkillTree();
    s.solution("CBD", new String[]{"BACDE", "CBADF", "AECB", "BDA"});
}*/
}

3중 for문은 실제로 사용하기 까다로워서 코딩문제에 잘 나오지 않을 것 같았는데, 구현하려다 보니 자동적으로 3중 for문 까지 가는 것 같다.

더 간단한 방법은 없을까

이 문제는 어떤 인덱스를 기준으로 서로의 스킬을 비교할 것인지 가장 중요한 것 같다.

그에 따라서 반복문이 도는 횟수가 달라지기 때문이다.

위의 코드를 보면

skill_trees에 들어있는 인덱스를 모두 비교해야 하기 때문에 treeLength만큼 처음 반복문을 돌려준다.

그 다음은 각 인덱스의 한 글자씩을 분리해야만 skill의 값과 비교할 수 있다. skill도 마찬가지로 비교하기위해 한 글자씩 분리해야한다.

나는 charAt()을 사용해 값을 가져왔다. tree의 각 인덱스의 길이만큼 반복문을 돌려 skill과 비교할 준비를 한다.

비교값이 같은지 판단하기위해 skillIndex와 boolean flag를 사용했다.

스킬의 한 글자와, 스킬트리 인덱스의 한 글자가 같은 경우,

스킬트리에 맞는 글자이기 때문에 Index를 ++해준다. => 다음 글자를 비교하기 위해서

true인 경우를 카운트한다.

다른 풀이

public int solution(String skill, String[] skill_trees) {
        int answer = 0;
        int treeLength = skill_trees.length;

        for(int i=0; i<treeLength; i++){
            boolean flag = true;
            String[] skills = skill_trees[i].split("");
            int cnt=0;

            int skillsLen = skills.length;
            for(int j=0; j<skills.length; j++){
                if(cnt < skill.indexOf(skills[j])){
                    flag = false;
                    break;
                }else if(cnt == skill.indexOf(skills[j])){
                    cnt++;
                }
            }

            if(flag){
                answer++;
            }
        }
        return answer;
    }