We're live on Product Hunt today! Use code PRODUCTHUNT26 for 20% off. Upvote us ↑
DSA Trainer
← All problems
PremiumTrieMedium

Implement Trie (Prefix Tree)

Asked at:AmazonGoogleMicrosoft

Design a trie (prefix tree) that supports inserting a word, searching for a word, and checking if any word in the trie starts with a given prefix.

This is a premium problem.

Unlock Implement Trie (Prefix Tree) with both full courses, DSA and System Design, and every other premium problem.

Problem

A trie (pronounced as 'try') or prefix tree is a tree data structure used to efficiently store and retrieve keys in a dataset of strings. Implement the Trie class with insert, search, and startsWith methods.

Input

A series of Trie operations: `insert(word)`, `search(word)`, `startsWith(prefix)`.

Output

Results for `search` and `startsWith` calls. `insert` returns nothing.

Examples

Input: trie.insert("apple"); trie.search("apple"); trie.search("app"); trie.startsWith("app"); trie.insert("app"); trie.search("app")

Output: true, false, true, true

The brute-force approach

Store all inserted words in a list. For search, check membership. For startsWith, check any word that starts with prefix.

words = []
insert(word) → words.append(word)
search(word) → word in words
startsWith(prefix) → any(w.startswith(prefix) for w in words)

O(m × n) for search and startsWith where m=word count, n=average length. Trie achieves O(L) for each operation where L is the word/prefix length.

Time: O(m×n)Space: O(m×n)

Spotting the pattern

This is a Trie problem. The key question to ask yourself:

What's the difference between search and startsWith in terms of what they check at the end of the traversal?

Answering that is where it clicks, and it's exactly what the guided walkthrough below builds with you: the pattern reasoning, a progressive hint ladder that never spoils the answer, a row-by-row dry run, the optimized solution, and an in-browser editor to run your code against real test cases.

Premium

Unlock the full guided solution

The Trie walkthrough for Implement Trie (Prefix Tree): the progressive hint ladder, a row-by-row dry run, the optimized code, and an in-browser runner, plus both full courses, DSA and System Design, and every other premium problem.

Already finished Big-O and Hash Maps free? This picks up right where the path leads: same format, harder patterns, the ones that show up in every FAANG-style screen.

What you get

  • The full DSA course: concept lessons, drills, graded tests, and cold-read capstones for every pattern
  • The full System Design course: 20 building-block units for the design round, included at no extra cost
  • Stop going blank on algorithm problems. The hint ladder walks you to the answer without spoiling it.
  • Actually internalize the trace: dry runs walk state row by row so it clicks before you write a line
  • Know when to reach for each pattern. Every problem names the trigger so you spot it next time.
  • Write and run JS or Python in the browser with real test cases on every problem
  • Access every new problem, unit, and pattern guide as they ship weekly
B
Big_Wolverine_7575Founding Member· unprompted on Reddit

“This app finally made it click. I tried NeetCode, CTCI, and YouTube for years and still couldn't solve Two Sum. The beginner mental models for the patterns are genuinely so helpful. THANK YOU.

Full access · both courses

Full access

$49 / year

or $9 / month

Now includes a C++ runner alongside JavaScript and Python.

LeetCode Premium is $159/year and doesn't teach you anything.

Less than one Udemy course, with guided problems instead of passive lectures.

One extra month of job searching costs more than a full year here.

  • Both courses, every unit: DSA + System Design
  • All 150 problems across 33 patterns, plus every new one
  • New problems and units weekly
  • Cancel anytime

Cancel anytime. Not useful within 7 days? Email for a full refund. Secured by Stripe.