문제 설명
선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.
예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.
위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크나 라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.
선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.
제한 조건
- 스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
- 스킬 순서와 스킬트리는 문자열로 표기합니다.
- 예를 들어, C → B → D 라면 CBD로 표기합니다
- 선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
- skill_trees는 길이 1 이상 20 이하인 배열입니다.
- skill_trees의 원소는 스킬을 나타내는 문자열입니다.
- skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.
입출력 예 설명
- BACDE: B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트립니다.
- CBADF: 가능한 스킬트리입니다.
- AECB: 가능한 스킬트리입니다.
- BDA: B 스킬을 배우기 전에 C 스킬을 먼저 배워야 합니다. 불가능한 스킬트리입니다.
문제 풀이
나는 Stack을 활용해서 풀었다. skill에 있는 값들을 하나씩 stack을 뒤에서 부터 넣었는데 queue로 푸는게 나아 보인다.
queue로 풀어도 되고 그렇게 할때는 순서대로 넣으면 된다.
stack에 넣고 skill에 있는 문자가 있을 경우 제일 위에있는 스택과 비교하여 같다면 pop을 해주고 같지 않다면 (순서가 다른거니까) break 시켜준다.
시간 복잡도 면에서는 많이 부족한 코드이다. 나중에 리팩토링을 해야 될 듯.
import java.util.*;
class Solution {
public int solution(String skill, String[] skill_trees) {
String[] skill_set = skill.split("");
Stack<String> stack = new Stack<String>();
int count = 0;
boolean isFail = false;
for (int i = 0; i < skill_trees.length; i++) {
//초기화
isFail = false;
stack.clear();
// 스택에 추가
for (int k = skill_set.length; k > 0; k--) {
stack.push(skill_set[k - 1]);
}
String[] temp = skill_trees[i].split("");
for (int j = 0; j < temp.length; j++) {
if (skill.indexOf(temp[j]) != -1) { // 포함여부 확인
if (stack.peek().equals(temp[j])) {
stack.pop();
} else {
isFail = true;
break;
}
}
}
if (!isFail) {
count++;
}
}
return count;
}
}
다른 사람 풀이
이 문제를 정규표현식을 이용하여 푼 신기한 발상이 있어서 가져왔다.
skill에 포함되지 않은 문자들을 다 제거한 뒤에 skill에 남아있는 문자가 포함되어 있는지 비교한다.
만약에 BDA 가 남아있을 경우 IndexOf로 남은 문자를 비교하면 스킬트리 순서도 함께 확인 가능하다.
import java.util.*;
class Solution {
public int solution(String skill, String[] skill_trees) {
int answer=skill_trees.length;
for(int i=0;i<skill_trees.length;i++){
if (skill.indexOf(skill_trees[i].replaceAll("[^"+skill+"]","")) !=0){
answer--;
}
}
return answer;
}
}
'Algorithm(알고리즘) > 프로그래머스 고득점 Kit' 카테고리의 다른 글
Java 프로그래머스 - 다리를 지나는 트럭 (Level 2 - Queue) (0) | 2021.01.10 |
---|---|
Java 프로그래머스 - 멀쩡한 사각형 (Level 2) (0) | 2021.01.10 |
Java 프로그래머스 - 124 나라의 숫자 Level2 (0) | 2021.01.08 |
Java 프로그래머스 - [2019 카카오] 실패율 Level1 (0) | 2021.01.07 |
Java 프로그래머스 - [2018 카카오 2번] 다트게임 Level1 (0) | 2021.01.06 |