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

 

 

■ 문제 설명

 

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

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

 

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

 

선행 스킬 순서 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

 

 

▶ 코드

 

기본적인 아이디어는 skill에 있는 ① 문자(char)이 있으면 queue에 넣고,

② queue에 있는 문자의 순서를 비교하는 것이다.

 

제출했는데 코드가 대부분의 테스트를 통과하지 못했다.

왜 그랬는지 몰랐는데 처음의 코드는 queue가 비어있으면 아래를 실행하지 않고 넘겼는데,

queue가 비어있으먼 선행해야할 skill이 없으므로 answer의 값을 증가시켜줘야한다. 

굉장히 간단하지만 잘 안보여서 찾는데 애 먹었는데 스터디원분이 찾아주셨다.

//Exception(스킬에 있는 문자가 아무것도 포함되어 있지 않은 경우)
//if(q.empty()) continue;
if(q.empty()) answer++;
        

 

#include <string>
#include <vector>
#include <queue>
#include <iostream>
using namespace std;

int solution(string skill, vector<string> skill_trees) {
    int answer = 0;
    queue<char> q;
    
    for(vector<string>::iterator iter = skill_trees.begin();iter!=skill_trees.end();iter++){
        string str = *iter;
        
        //스킬에 있는 문자면 queue에 push
        for(int i=0;i<str.length();i++){
            for(int j=0;j<skill.length();j++){
                if(str[i]==skill[j]) q.push(str[i]);
            }
        }
        
        //Exception(스킬에 있는 문자가 아무것도 포함되어 있지 않은 경우)
        if(q.empty()) answer++;
        
        //queue에 있는 순서 비교
        for(int i=0;i<skill.length();i++){
            if(skill[i] == q.front()) q.pop();
            else break;
            
            if(q.empty()){
                answer++;
                break;
            }
        }
        
        //queue 초기화
        while(!q.empty()) q.pop();
    }
    
    return answer;
}

 

※ 출처 : https://programmers.co.kr/learn/courses/30/lessons/49993

코딩테스트 연습 - 스킬트리 | 프로그래머스