[코딩테스트] 프로그래머스 - 스킬트리

프로그래머스의 스킬트리 문제를 풀어볼려고 한다.

 

 

문제 설명

 

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

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

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

선행 스킬 순서 skill과 유저들이 만든 스킬트리를 담은 배열 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 = 0;
        
        for( int i = 0; i < skill_trees.length ; i++){
            String check = "";
            
            for( int j=0 ; j < skill_trees[i].length() ; j++) {
                for(int z=0 ; z < skill.length() ; z++){
                    if(String.valueOf(skill_trees[i].charAt(j)).equals(String.valueOf(skill.charAt(z)))) {
                        check += String.valueOf(skill_trees[i].charAt(j));
                    }
                }
            }
            
            if(check.length() == 0){
                answer++;
            } else {
                for(int j = check.length() ; j < skill.length() ; j++) {
                    check += skill.charAt(j);
                }
                if(skill.equals(check)) {
                    answer++;
                }
            }
            
        }
        
        return answer;
    }
}

 

 

 

skill_tree 를 모두 돌고 거기서 한번더 해당 문자 x  스킬트리 문자수 만큼 for문이 도는 것이다.

 

n - skill_tree 갯수

m - skill 문자수

z -  스킬 문자 수

시간복잡도로 O(n) * O(m) * O(z) 로 최악의 경우에는 O(n^3) 이 되는 굉장히 안좋은 코드를 작성을 했다.

 

 

더 빠르게 수정해보기

 

 

for을 줄일 수 있는 생각을 해보았다.

 

여기서 생각해낸 방법이 주어진 스킬트리를 제외하고 모두 빈값으로 만든 후, 순서를 계산하자는 것이다.

 

 

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

        for (String s : skill_trees) {
            String s2 = s.replaceAll("[^"+skill+"]", "");  // BCD, CBD, CB, BD  로 변환
            if(skill.startsWith(s2)) answer++;
        }

        return answer;
 }

 

java의 문자열 메소드 중에 startsWith 를 사용하여 앞의 있는지를 체크를 하였다.

 

 

첫 케이스 빼고는 다른 케이스에서 확실이 빨라진것을 느껴진것을 알게되었다.

 

 

- 리팩토링 하기

 

 

소스를 더 간결하게 하기 위해 소스를 수정을 하였다.

java 8에 추가된 stream으로 만든 후, 바로 반환되게 수정하였다.

 

import java.util.Arrays;

public int solution(String skill, String[] skill_trees) {
        return (int) Arrays.stream(skill_trees)
                .map(s -> s.replaceAll("[^"+skill+"]", ""))
                .filter(s -> skill.startsWith(s))
                .count();
    }