프로그래머스] C++ 2020 KAKAO BLIND RECRUITMENT - 가사 검색(Level 4)

해당 문제는 프로그래머스 코딩테스트 연습에 있는 문제입니다.

아래 링크를 통해 풀 수 있습니다.

 

 

https://programmers.co.kr/learn/courses/30/lessons/60060#

 

 

 

 

[본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.]

친구들로부터 천재 프로그래머로 불리는 프로도는 음악을 하는 친구로부터 자신이 좋아하는 노래 가사에 사용된 단어들 중에 특정 키워드가 몇 개 포함되어 있는지 궁금하니 프로그램으로 개발해 달라는 제안을 받았습니다.
그 제안 사항 중, 키워드는 와일드카드 문자중 하나인 '?'가 포함된 패턴 형태의 문자열을 뜻합니다. 와일드카드 문자인 '?'는 글자 하나를 의미하며, 어떤 문자에도 매치된다고 가정합니다. 예를 들어 "fro??"는 "frodo", "front", "frost" 등에 매치되지만 "frame", "frozen"에는 매치되지 않습니다.

가사에 사용된 모든 단어들이 담긴 배열 words와 찾고자 하는 키워드가 담긴 배열 queries가 주어질 때, 각 키워드 별로 매치된 단어가 몇 개인지 순서대로 배열에 담아 반환하도록 solution 함수를 완성해 주세요.

가사 단어 제한사항

• words의 길이(가사 단어의 개수)는 2 이상 100,000 이하입니다.
• 각 가사 단어의 길이는 1 이상 10,000 이하로 빈 문자열인 경우는 없습니다.
• 전체 가사 단어 길이의 합은 2 이상 1,000,000 이하입니다.
• 가사에 동일 단어가 여러 번 나올 경우 중복을 제거하고 words에는 하나로만 제공됩니다.
• 각 가사 단어는 오직 알파벳 소문자로만 구성되어 있으며, 특수문자나 숫자는 포함하지 않는 것으로 가정합니다.

검색 키워드 제한사항

• queries의 길이(검색 키워드 개수)는 2 이상 100,000 이하입니다.
• 각 검색 키워드의 길이는 1 이상 10,000 이하로 빈 문자열인 경우는 없습니다.
• 전체 검색 키워드 길이의 합은 2 이상 1,000,000 이하입니다.
• 검색 키워드는 중복될 수도 있습니다.
• 각 검색 키워드는 오직 알파벳 소문자와 와일드카드 문자인 '?' 로만 구성되어 있으며, 특수문자나 숫자는 포함하지 않는 것으로 가정합니다.
• 검색 키워드는 와일드카드 문자인 '?'가 하나 이상 포함돼 있으며, '?'는 각 검색 키워드의 접두사 아니면 접미사 중 하나로만 주어집니다.
  • 예를 들어 "??odo", "fro??", "?????"는 가능한 키워드입니다.
  • 반면에 "frodo"('?'가 없음), "fr?do"('?'가 중간에 있음), "?ro??"('?'가 양쪽에 있음)는 불가능한 키워드입니다.

입출력 예

words	queries	result
["frodo", "front", "frost", "frozen", "frame", "kakao"]	["fro??", "????o", "fr???", "fro???", "pro?"]	[3, 2, 4, 1, 0]

입출력 예에 대한 설명

• "fro??"는 "frodo", "front", "frost"에 매치되므로 3입니다.
• "????o"는 "frodo", "kakao"에 매치되므로 2입니다.
• "fr???"는 "frodo", "front", "frost", "frame"에 매치되므로 4입니다.
• "fro???"는 "frozen"에 매치되므로 1입니다.
• "pro?"는 매치되는 가사 단어가 없으므로 0 입니다.

 

 

 

 

전형적인 Trie알고리즘으로 푸는 문제입니다.

 

고려해야할 점은 다음과 같습니다.

 

1. '?'가 앞에 있는 경우

2. '?'를 만났을 때의 탐색 방법

3. '?'로만 이뤄진 문자열

 

 

저는 두개의 Trie를 선언했습니다.

하나는 words를 그대로 넣고 다른 하나는 words의 문자열을 뒤집어서 넣었습니다.

이렇게 되면 '?'가 앞에있든 뒤에있든 모두 탐색이 가능합니다.

 

'?'를 만났을 때의 처리는 다음과 같이 했습니다.

frodo를 저장한다고 가정한다면,

frodo의 길이를 map에 저장.

그 후 탐색이 이뤄질때마다 단어가 하나씩 줄어든 길이를 map에 저장했습니다.

ex) frodo = 5, rodo = 4, odo = 3, do = 2, o = 1

이렇게 된다면 각 단어 하나씩 저장될 때마다 남은 길이를 저장하게 되며 '?'의 갯수와 동일한 길이를 반환해주면 됩니다.

 

 "fr???"를 찾는다고 가정한다면

fr까지 찾은 후 '?'부터는 map에 저장되어 있는 단어크기의 갯수를 저장 후 return했습니다.

 

#include <string>
#include <vector>
#include <iostream>
#include <map>
#include <algorithm>

using namespace std;
int w_count = 0;

class Trie {
private:
    Trie * child[26]{};
    map<int, int> m;
    bool check = false;
public:
    void insert(const char * word, int n, int w_size){
        if(*word == '\0'){ return; }
        int next = *word - 'a';
        m[w_size]++;
        check = true;
        
        if(child[next] == 0){
            child[next] = new Trie();
            child[next]->insert(word + 1, n + 1, w_size - 1);
        }else
            child[next]->insert(word + 1, n + 1, w_size - 1);
        
        return;
    }
    
    void find(const char * word, int word_len){
        if(*word == '\0'){
            if(check == false)
                { w_count++; return; }
            return;
        }

        if(*word == '?'){ w_count = m[word_len]; return; }
        int next = *word - 'a';
        if(child[next] != 0){ child[next]->find(word + 1, word_len - 1); }
        
        return;
    }
    ~Trie(){
        for(int i = 0; i < 26; i++) delete child[i];
        map<int, int>().swap(m);
    }
};

vector<int> solution(vector<string> words, vector<string> queries) {
    vector<int> answer;
    Trie trie, r_trie;
    
    for(int i = 0; i < words.size(); i++){
        trie.insert(words[i].c_str(), 1, words[i].size());
        reverse(words[i].begin(), words[i].end());
        r_trie.insert(words[i].c_str(), 1, words[i].size());
    }
    
    for(int i = 0; i < queries.size(); i++){
        if(queries[i][0] == '?'){
            reverse(queries[i].begin(), queries[i].end());
            r_trie.find(queries[i].c_str(), queries[i].size());
        }
        else
            trie.find(queries[i].c_str(), queries[i].size());
        
        if(w_count == 0) answer.emplace_back(0);
        else{ answer.emplace_back(w_count); w_count = 0; }        
    }
    
    return answer;
}