Why Count Words with Given Prefix in DSA Go?

The Big Idea

What if you could find all words starting with your letters instantly, no matter how big the list?

The Scenario

Imagine you have a huge list of words, like a phone book or a dictionary, and you want to find how many words start with a certain set of letters, like "pre".

Doing this by checking each word one by one is like looking through every page of a phone book to find names starting with "Pre"--it takes a lot of time.

The Problem

Manually scanning each word is slow and tiring, especially if the list is very long.

It's easy to make mistakes, like missing some words or counting wrong.

This method wastes time and effort, making it hard to get quick answers.

The Solution

Using a special structure called a Trie, we can quickly jump to the part where words with the prefix start.

This way, we don't check every word, only the relevant ones, making counting fast and easy.

Before vs After

✗ Before

count := 0
for _, word := range words {
    if strings.HasPrefix(word, prefix) {
        count++
    }
}

✓ After

count := trie.CountWordsWithPrefix(prefix)

What It Enables

This lets us instantly find how many words start with any prefix, even in huge word lists.

Real Life Example

When you type in a search box, the system suggests words starting with what you typed. This fast prefix counting helps show those suggestions quickly.

Key Takeaways

Manual checking is slow and error-prone.

Trie structure speeds up prefix counting.

Enables fast search suggestions and word lookups.

Practice

1. What does the problem "Count Words with Given Prefix" ask you to do?

easy

A. Count the total number of characters in all words

B. Find the longest word in a list

C. Count how many words in a list start with a specific prefix

D. Sort the words alphabetically

Solution

1. Step 1: Understand the problem statement

   The problem asks to find how many words begin with a given prefix string.

2. Step 2: Identify the key operation

   We need to check each word and count those that start with the prefix.

3. Final Answer:

   Count how many words in a list start with a specific prefix -> Option C

4. Quick Check:

   Counting words by prefix = Count how many words in a list start with a specific prefix [OK]

Hint: Look for words starting exactly with the prefix [OK]

Common Mistakes:

• Confusing prefix with substring anywhere in word
• Counting total words instead of filtered ones
• Ignoring case sensitivity if required

2. Which of the following Go code snippets correctly checks if a word starts with a given prefix pref?

easy

A. strings.HasPrefix(word, pref)

B. strings.Contains(word, pref)

C. word[:len(pref)] == pref

D. word == pref

Solution

1. Step 1: Recall Go standard library functions

   strings.HasPrefix(word, pref) returns true if word starts with pref.

2. Step 2: Compare other options

   strings.Contains checks anywhere, slicing may panic if prefix longer, equality checks full word.

3. Final Answer:

   strings.HasPrefix(word, pref) -> Option A

4. Quick Check:

   Use HasPrefix for prefix check [OK]

Hint: Use strings.HasPrefix for safe prefix check [OK]

Common Mistakes:

• Using strings.Contains which checks anywhere in word
• Slicing without length check causing runtime panic
• Comparing whole word instead of prefix

3. What is the output of the following Go code?

package main
import (
  "fmt"
  "strings"
)
func countPrefix(words []string, prefix string) int {
  count := 0
  for _, w := range words {
    if strings.HasPrefix(w, prefix) {
      count++
    }
  }
  return count
}
func main() {
  words := []string{"apple", "app", "application", "banana", "apply"}
  fmt.Println(countPrefix(words, "app"))
}

medium

A. 4

B. 2

C. 3

D. 5

Solution

1. Step 1: Identify words starting with "app"

   Words: "apple", "app", "application", "banana", "apply". "banana" does not start with "app".

2. Step 2: Count matching words

   Matching words: "apple", "app", "application", "apply" -> 4 words.

3. Final Answer:

   4 -> Option A

4. Quick Check:

   Count of words starting with "app" = 4 [OK]

Hint: Count words starting exactly with prefix using HasPrefix [OK]

Common Mistakes:

• Counting "banana" incorrectly
• Off-by-one error in counting
• Misreading prefix length

4. The following Go code is intended to count words starting with a prefix but causes a runtime error. What is the error?

func countPrefix(words []string, prefix string) int {
  count := 0
  for _, w := range words {
    if w[:len(prefix)] == prefix {
      count++
    }
  }
  return count
}

medium

A. No error, code works fine

B. Index out of range panic if word shorter than prefix

C. Logical error: counts words containing prefix anywhere

D. Syntax error due to missing import

Solution

1. Step 1: Analyze slicing operation

   The code slices w[:len(prefix)] without checking if w is long enough.

2. Step 2: Identify runtime error cause

   If w is shorter than prefix, slicing causes index out of range panic.

3. Final Answer:

   Index out of range panic if word shorter than prefix -> Option B

4. Quick Check:

   Slicing without length check causes panic [OK]

Hint: Check word length before slicing prefix [OK]

Common Mistakes:

• Ignoring length check before slicing
• Assuming code compiles without imports
• Confusing prefix check with substring check

5. You have a list of words and want to count how many start with each prefix from a list of prefixes. Which approach is best for efficient counting in Go?

hard

A. Use strings.Contains to check prefix in each word

B. For each prefix, loop over all words and count matches

C. Sort words and use binary search for each prefix

D. Build a trie from words, then query counts for each prefix

Solution

1. Step 1: Understand the problem scale

   Counting for many prefixes over many words needs efficiency.

2. Step 2: Compare approaches

   Looping over all words for each prefix is slow (O(P*W)). Sorting + binary search helps but trie allows prefix counting efficiently.

3. Step 3: Choose best data structure

   Trie stores words by prefix, enabling fast prefix count queries.

4. Final Answer:

   Build a trie from words, then query counts for each prefix -> Option D

5. Quick Check:

   Trie enables efficient prefix counting [OK]

Hint: Use trie for fast prefix count queries [OK]

Common Mistakes:

• Using nested loops causing slow performance
• Using strings.Contains which is incorrect for prefix
• Ignoring data structure benefits