[ 프로그래머스 - Java & Kotlin ] 스킬트리 ( 자바 & 코틀린 )

( Summer/Winter Coding(~2018) / 스킬 트리 )

[문제]

문제 설명

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

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

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

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

제한사항

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

---

[코드]

자바

class Solution {
    public int solution(String skill, String[] skill_trees) {
        int answer = 0;
        
        for(String str : skill_trees){
            int idx = -1;
            boolean flag = true;
            for(char c : str.toCharArray()){
                // 현재 스킬 인덱스 저장
                int now = skill.indexOf(c);
                // 현재 스킬이 skill에 있을 경우
                if(now != -1){
                    // skill에 선행 스킬을 사용 안했거나 다음 스킬 이상으로 건너 뛴 경우 반복 중단하고 flag false로 초기화
                    if(now <= idx || now != idx + 1){
                        flag = false;
                        break;
                    }
                    idx = now;
                }
            }
            // 모든 스킬을 정상적으로 사용했을 경우 answer++
            if(flag) answer++;
        }
        
        return answer;
    }
}

코틀린

fun solution(skill:String, skill_trees:Array<String>):Int{
    var answer = 0

    for(str in skill_trees){
        var idx = -1
        var flag = true
        for(c in str){
            // 현재 스킬 인덱스 저장
            val now = skill.indexOf(c)
            // 현재 스킬이 skill에 있을 경우
            if(now != -1){
                // skill에 선행 스킬을 사용 안했거나 다음 스킬 이상으로 건너 뛴 경우 반복 중단하고 flag false로 초기화
                if(now <= idx || now != idx + 1) {
                    flag = false
                    break
                }
                idx = now
            }
        }
        // 모든 스킬을 정상적으로 사용했을 경우 answer++
        if(flag) answer++
    }

    return answer
}

문제 링크

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

제가 잘못 알고 있거나 잘못된 부분이 있을 경우 알려주시고 추가로 궁금한 점 있으신 분들도 댓글이나 메일 주시면 성실히 답변해 드리겠습니다.🧑🏻‍💻

감사합니다~😄