Search⌘ K
AI Features

Solution: Map Sum Pairs

Explore how to implement a MapSum data structure that supports inserting string keys with integer values and retrieving sum of values for keys sharing a common prefix. Understand using a trie for efficient prefix aggregation and how to handle updates without double counting. By the end, you'll be able to build a trie-based solution that efficiently computes prefix sums in coding interviews.

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