Trie (Insert and Search)

Hello Everyone,

Trie is an efficient information re Trie val data structure. Using Trie, search complexities can be brought to optimal limit (key length). If we store keys in binary search tree, a well balanced BST will need time proportional to M * log N , where M is maximum string length and N is number of keys in tree. Using Trie, we can search the key in O(M) time. However the penalty is on Trie storage requirements.

Every node of Trie consists of multiple branches. Each branch represents a possible character of keys. We need to mark the last node of every key as end of word node. A Trie node field isEndOfWord is used to distinguish the node as end of word node. A simple structure to represent nodes of the English alphabet can be as following,
// Trie nodestruct TrieNode{ struct TrieNode *children[ALPHABET_SIZE];

// isEndOfWord is true if the node // represents end of a word bool isEndOfWord;};

Inserting a key into Trie is a simple approach. Every character of the input key is inserted as an individual Trie node. Note that the children is an array of pointers (or references) to next level trie nodes. The key character acts as an index into the array children . If the input key is new or an extension of the existing key, we need to construct non-existing nodes of the key, and mark end of the word for the last node. If the input key is a prefix of the existing key in Trie, we simply mark the last node of the key as the end of a word. The key length determines Trie depth.

Searching for a key is similar to insert operation, however, we only compare the characters and move down. The search can terminate due to the end of a string or lack of key in the trie. In the former case, if the isEndofWord field of the last node is true, then the key exists in the trie. In the second case, the search terminates without examining all the characters of the key, since the key is not present in the trie.

Insert and search costs O(key_length) , however the memory requirements of Trie is O(ALPHABET_SIZE * key_length * N) where N is number of keys in Trie. There are efficient representation of trie nodes (e.g. compressed trie, ternary search tree, etc.) to minimize memory requirements of trie.

// C++ implementation of search and insert

// operations on Trie

#include <bits/stdc++.h>

using namespace std;

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 ;

}

// Returns true if key presents in trie, else

// false

bool 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 false ;

pCrawl = pCrawl->children[index];

}

return (pCrawl != NULL && pCrawl->isEndOfWord);

}

// Driver

int main()

{

// Input keys (use only 'a' through 'z'

// and lower case)

string keys[] = { "the" , "a" , "there" ,

"answer" , "any" , "by" ,

"bye" , "their" };

int n = sizeof (keys)/ sizeof (keys[0]);

struct TrieNode *root = getNode();

// Construct trie

for ( int i = 0; i < n; i++)

insert(root, keys[i]);

// Search for different keys

search(root, "the" )? cout << "Yes\n" :

cout << "No\n" ;

search(root, "these" )? cout << "Yes\n" :

cout << "No\n" ;

return 0;

}

Output :

the — Present in trie
these — Not present in trie
their — Present in trie
thaw — Not present in trie