Problem · Prefix Sum

Find Maximum Total Amount (SDE I, Fungible :)

MediumAmazonNEW GRADOA
See Amazon hiring insights

In Amazon's financial team, an analyst is dealing with an infinite number of bags arranged in a line, each numbered from 1 to infinity. The task is to gather information about the amount of money in these bags, presented in the form of continuous segments. The objective is to select consecutive bags in such a way that the total amount of money in these bags is maximized.

The continuous segments provided to represent the amount of money in each bag do not intersect. Additionally, any bag included within a segment contains some amount of money, while bags not included in any segment are considered to have zero money.

The problem statement continues
Pro
Examples
01 · Example 1
segment = [[1, 4, 2], [6, 6, 5], [7, 7, 7], [9, 10, 1]]
k = 5
return = 16

The amount of money in each bag is:

Bag 1 2 3 4 5 6 7 8 9 10
Money 2 2 2 2 0 5 7 0 1 1

All other bags have zero money.

Let's try different consecutive bags of size k:

Bags Total Money
[1 - 5]2 + 2 + 2 + 2 + 0 = 8
[2 - 6]2 + 2 + 2 + 0 + 5 = 11
[3 - 7]2 + 2 + 0 + 5 + 7 = 16
[4 - 8]2 + 0 + 5 + 7 + 0 = 14
[5 - 9]0 + 5 + 7 + 0 + 1 = 13
[6 - 10]5 + 7 + 0 + 1 + 1 = 14

The subsegment starting from the third bag and ending at the seventh bag has the maximum total amount of money, hence the answer is 16.

2 original screenshots from the interview report
Pro
FastPrep Pro
Reported in 1 Amazon interview this week

Unlock the full problem

  • Full problem statement and constraints
  • Worked examples, explained
  • Guided hints and editorial
  • Run your code on real test cases
$9/month

Pro subscription, billed yearly — or $19 month-to-month. Cancel anytime.

Free plan — 2 of 2 free unlocks used this week
CodePython 3
Run and Submit unlock with Pro
FastPrep Pro
Reported in 1 Amazon interview this week

Unlock the full problem

  • Full problem statement and constraints
  • Worked examples, explained
  • Guided hints and editorial
  • Run your code on real test cases
$9/month

Pro subscription, billed yearly — or $19 month-to-month. Cancel anytime.

Free plan — 2 of 2 free unlocks used this week