算法导论—Trie树

struct TrieNode
{
int count;
TrieNode* next[26];
bool exist;
TrieNode() :count(0), exist(false);
};

void Insert(TrieNode* root, string word)
{
TrieNode* node = root;
for (int i = 0; i < word.length(); i++)
{
int id = word[i] - 'a';
if (node->next[id] == NULL)
{
node->next[id] = new TrieNode();
}
node = node->next[id];
node->count += 1;
}
node->exist = true;
}

int Search(TrieNode* root, string word)
{
TrieNode* node = root;
for (int i = 0; i < word.length(); i++)
{
int id = word[i] - 'a';
node = node->next[id];
if (node == NULL)
return 0;
}
return node->count;
}

int main(void)
{
TrieNode* root = new TrieNode();
int n;
string s;
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> s;
Insert(root, s);
}
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> s;
cout << Search(root, s) << endl;
}
return 0;
}