Search⌘ K
AI Features

Solution: Map Sum Pairs

Explore how to design a data structure that supports inserting and updating key-value pairs with string keys and integer values. Understand the use of a trie combined with a hash map to efficiently compute the sum of all values for keys sharing a given prefix. Learn to handle updates without double counting by computing deltas and propagating them through the trie nodes. This lesson helps you master prefix-based retrieval and value updates with optimal time complexity.

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 ...