def trie_search_surname(surname): book_list = load_books(file) book_list = book_list[1:] surnames = [] for book in book_list: author = book[3] author_last_name = author.split(" ")[-1] surnames.append(author_last_name) root = TrieNode() root.insert_many(surnames) return root.find(req_title)
def trie_search_id(req_title): book_list = load_books(file) titles = [] for i in range(len(book_list)): if i == 0: continue else: book = book_list[i] title = "".join(book[2:len(book) - 2]) titles.append(title) root = TrieNode() root.insert_many(titles) return root.find(req_title)