[프로그래머스/C++] 스킬트리

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

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

스킬트리

문제

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

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

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

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

제한사항

스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.
스킬 순서와 스킬트리는 문자열로 표기합니다. 예를 들어, C → B → D 라면 "CBD"로 표기합니다
선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.
skill_trees는 길이 1 이상 20 이하인 배열입니다.
skill_trees의 원소는 스킬을 나타내는 문자열입니다.
skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

 

풀이

입출력 예시

 

풀이

중간에 다른 스킬이 섞이는 것은 상관없지만, skill에 주어진 순서는 따라야 한다는 것에 주의해야한다. 3번째 경우인 AECB는 D스킬을 배우지는 않았지만 skill에 주어진 순서대로 배웠으므로 가능한 스킬트리이다.

#include <string>
#include <vector>

using namespace std;

int solution(string skill, vector<string> skill_trees) {
    int answer = 0;
    bool check_skill = true;
    string sk;

    for (const auto& trees : skill_trees) {
        for (const auto& k : trees) {
            if (skill.find(k) != string::npos) { // k가 skill에 있는 스킬일 경우에만 저장
                sk.push_back(k);
            }
        }

        for (int i = 0; i < sk.length(); i++) {
            if (sk[i] != skill[i]) { // 순서가 다르면 루프를 멈춘다.
                check_skill = false;
                break;
            }
        }

        if (check_skill) { // 순서가 맞을 때만 증가
            answer++;
        }
        check_skill = true;
        sk.clear();
    }
    
    return answer;
}

먼저 스킬트리가 순서대로 되어있는지 확인할 문자열 sk를 선언해줬다. sk에는 skill에 해당되는 스킬만 저장이 된다. 이어서 sk와 skill의 순서가 같은지를 비교한다. 이때, 3번째 경우가 있기 때문에 compare() 대신에 위치를 하나하나 비교해줬다.

 

결과