Search⌘ K
AI Features

Solution: Map Sum Pairs

Understand how to design a data structure using a trie that supports inserting string keys with integer values and calculates the sum of all values for keys starting with a given prefix. Learn to update values efficiently by handling key updates and querying prefix sums without scanning all keys. This lesson offers a clear implementation and explains the time and space complexity of using a trie combined with a hash map for optimized prefix sum operations.

Statement

Design a data structure that supports the following operations:

  1. Insert a key-value pair:

    1. Each key is a string, and each value is an integer.

    2. If the key already exists, update its value to (overriding the previous value).

  2. Return the prefix sum:

    1. Given a string, prefix, return the total sum of all values associated with keys that start with this prefix.

To accomplish this, implement a class MapSum:

  • Constructor: Initializes the object.

  • void insert (String key, int val): Inserts the key-value pair into the data structure. If the key already exists, its value is updated to the new one.

  • int sum (String prefix): Returns the total sum of values for all keys that begin with the specified prefix.

Constraints:

  • 11 \leq key.length, prefix.length ...