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

level 2

Posted by dongjune on November 24, 2020

문제

스킬 트리

풀이

skill_trees의 첫번째 원소가 “BACDE”, skill 이 “CBD”라고 해보자.

우선 BACDE 를 한 문자씩 순회하며 CBD중 한 가지이면 temp vector에 추가해준다.
이 경우에 temp는 {B, C, D} 가 된다.
temp가 CBD의 순서를 지킨다면 이 스킬트리는 옳은 스킬트리이다. 하지만 이 경우는 {B,C,D} 이므로 옳지 않다.
이번에는 skill tree가 “ACBE” 라면 temp vector 는 {C,B}가 된다. 이 경우는 CBD의 순서를 지키므로 옳은 스킬트리이다.
이제 이 알고리즘을 소스 코드를 통해 구현해보자.

소스 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#include <string>
#include <vector>

using namespace std;

int solution(string skill, vector<string> skill_trees) {
    int answer = 0;
    // skill_trees 순회
    for(const auto &st:skill_trees)
    {
        
        vector<int> temp;

        // skill tree를 한 문자씩 순회
        for(const auto &s:st){
            // skill tree의 문자가 skill의 문자를 포함하면 temp에 추가해준다.
            for(const auto &r:skill){
                if(s==r){
                    temp.push_back(s);
                }
            }
        }
        bool flag=true;
        // temp가 skill의 순서를 따르지 않으면 flag=false
        for(int i=0;i<temp.size();i++){
            if(temp[i]!=skill[i]){
                flag=false;
                break;
            }
        }
        if(flag)
            answer++;
    }
    return answer;
}