공부기록/프로그래머스

[프로그래머스] 스킬트리

메델 2024. 2. 3. 15:47
import java.util.*; 

class Solution {
    public int solution(String skill, String[] skill_trees) {
        int answer = 0;
        
        for(int i=0; i<skill_trees.length; i++){
            skill_trees[i] = skill_trees[i].replaceAll("[^"+skill+"]", "");
        }
        
        HashSet<String> set  = new HashSet<>();
        
        for(int i=0; i<skill.length(); i++){
            set.add(skill.substring(0, i+1));
        }
        
        for(int i=0; i<skill_trees.length; i++){
            if(set.contains(skill_trees[i])){
                answer++;
            }
            if(skill_trees[i].equals("")){
                answer++;
            }
        }

        
        return answer;
    }
}

 

처음 풀었을때 틀렸는데 빈문자열의 경우를  고려하지 못해서 틀렸다..