Commit 443dac2

authored

Fixing Trie.py

Handling non-alphabet user input if not between [a-z].

1 parent e27f6c7 commit 443dac2Copy full SHA for 443dac2

File tree

-2

lines changed

-2

lines changed

Lines changed: 7 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-`
`2`	`1`	`class TrieNode(object):`
`3`	`2`	`def __init__(self):`
`4`	`3`	`self.children = [] #will be of size = 26`
`@@ -17,6 +16,9 @@ def insert(self, root, key):`
`17`	`16`	`pCrawl = root`
`18`	`17`	`for i in key:`
`19`	`18`	`index = ord(i)-97`
	`19`	`+ if index > 25 or index < 0:`
	`20`	`+ print("Small alphabet charchters only allowed")`
	`21`	`+ return`
`20`	`22`	`if(pCrawl.children[index] == None):`
`21`	`23`	`# node has to be initialised`
`22`	`24`	`pCrawl.children[index] = self.getNode()`
`@@ -28,6 +30,9 @@ def search(self, root, key):`
`28`	`30`	`pCrawl = root`
`29`	`31`	`for i in key:`
`30`	`32`	`index = ord(i)-97`
	`33`	`+ # handling non alphabet characters`
	`34`	`+ if index > 25 or index < 0:`
	`35`	`+ return False`
`31`	`36`	`if(pCrawl.children[index] == None):`
`32`	`37`	`return False`
`33`	`38`	`pCrawl = pCrawl.children[index]`
`@@ -45,4 +50,4 @@ def main():`
`45`	`50`	`else:`
`46`	`51`	`print("The given word doesnt exist")`
`47`	`52`	`if __name__ == '__main__':`
`48`		`- main()`
	`53`	`+ main()`

Comments

(0)