[백준] 14426 접두사 찾기
백준 14426 접두사 찾기
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
// C++ implementation of search and insert
// operations on Trie
const int ALPHABET_SIZE = 26;
// trie node
struct TrieNode{
struct TrieNode *children[ALPHABET_SIZE];
// isEndOfWord is true if the node represents
// end of a word
bool isEndOfWord;
};
// Returns new trie node (initialized to NULLs)
struct TrieNode *getNode(void){
struct TrieNode *pNode = new TrieNode;
pNode->isEndOfWord = false;
for (int i = 0; i < ALPHABET_SIZE; i++)
pNode->children[i] = NULL;
return pNode;
}
// If not present, inserts key into trie
// If the key is prefix of trie node, just
// marks leaf node
void insert(struct TrieNode *root, string key){
struct TrieNode *pCrawl = root;
for (int i = 0; i < key.length(); i++)
{
int index = key[i] - 'a';
if (!pCrawl->children[index])
pCrawl->children[index] = getNode();
pCrawl = pCrawl->children[index];
}
// mark last node as leaf
pCrawl->isEndOfWord = true;
}
//트리에 key가 존재하는 경우 1
//트리에 key가 다른 단어의 접두사로 존재하는 경우 0
//트리에 key가 존재하지 않는 경우 -1
int search(struct TrieNode *root, string key){
struct TrieNode *pCrawl = root;
for (int i = 0; i < key.length(); i++)
{
int index = key[i] - 'a';
if (!pCrawl->children[index])
return -1;
pCrawl = pCrawl->children[index];
}
return pCrawl->isEndOfWord;
}
// Driver
int main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
int n, m;
cin >> n >> m;
struct TrieNode *root = getNode();
for (int i = 0; i < n; ++i) {
string input;
cin >> input;
insert(root, input);
}
//포함되어 있는 문자열 중 적어도 하나의 접두사의 수
int cnt = 0;
for (int i = 0; i < m; ++i) {
string input;
cin >> input;
if (search(root, input) != -1) ++cnt;
}
cout << cnt;
return 0;
}
Author And Source
이 문제에 관하여([백준] 14426 접두사 찾기), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@sunjoo9912/백준-14426-접두사-찾기저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)