Advertisement
xxZeus

good set trie

Sep 27th, 2021
908
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.61 KB | None | 0 0
  1. class Trie:
  2.     def __init__(self):
  3.         self.nodes = dict()
  4.         self.end = False
  5.    
  6.     def insert(self, s, i):
  7.         if i == len(s) - 1:
  8.             self.end = True
  9.             if len(self.nodes):
  10.                 return False
  11.             return True
  12.         if self.end:
  13.             return False
  14.         if s[i] not in self.nodes:
  15.             self.nodes[s[i]] = Trie()
  16.         return self.nodes[s[i]].insert(s, i+1)
  17.    
  18. w = input().split()
  19. flg = True
  20. t = Trie()
  21. for x in w:
  22.     v = t.insert(x, 0)
  23.     if not v:
  24.         flg = False
  25.         break
  26.  
  27. print("Good" if flg else "not good")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement