DSA Trainer
Heaps and Priority Queues
← All problems
PremiumTwo HeapsHard

IPO

Asked at:Amazon

You can complete at most k projects before an IPO. Each project has a profit and a capital requirement. Starting with initial capital w, pick at most k projects to maximize your final capital. You can only start a project if your current capital meets its requirement.

This is a premium problem.

Unlock IPO with the full guided learning path and every other premium problem.

Problem

LeetCode will start its IPO soon. To maximize its total capital, it can finish at most k distinct projects before the IPO. Each project has a pure profit and a minimum capital required. Given k, w (initial capital), and two arrays profits and capital, return the maximized total capital after finishing at most k distinct projects.

Input

Integers k (max projects) and w (starting capital), plus arrays `profits` and `capital` where project i has profit profits[i] and requires capital[i] to start.

Output

The maximum capital you can accumulate after completing at most k projects.

Examples

Input: k = 2, w = 0, profits = [1,2,3], capital = [0,1,1]

Output: 4

Start with w=0. Only project 0 is affordable (capital=0). Complete it: w=1. Now projects 1 and 2 are affordable. Pick project 2 (profit=3): w=4. Done after 2 projects.

Input: k = 3, w = 0, profits = [1,2,3], capital = [0,1,2]

Output: 6

Complete all three in order. Each completion unlocks the next.

The brute-force approach

At each step, scan all unfinished projects, filter to those you can afford, pick the one with the highest profit, add its profit to w. Repeat up to k times.

for _ in range(k):
    best = max((p for c,p in zip(capital,profits) if c <= w and not done),
               default=None)
    if best is None: break
    w += best
    mark best as done

O(k × n) — each of the k rounds scans all n projects.

Time: O(k × n)Space: O(n)

Spotting the pattern

This is a Two Heaps problem. The key question to ask yourself:

At each step, which projects become available, and among those, which should you always pick?

Answering that is where it clicks, and it's exactly what the guided walkthrough below builds with you: the pattern reasoning, a progressive hint ladder that never spoils the answer, a row-by-row dry run, the optimized solution, and an in-browser editor to run your code against real test cases.

Premium

Unlock the full guided solution

The Two Heaps walkthrough for IPO: the progressive hint ladder, a row-by-row dry run, the optimized code, and an in-browser runner, plus the full guided learning path and every other premium problem.

Already finished Big-O and Hash Maps free? This picks up right where the path leads: same format, harder patterns, the ones that show up in every FAANG-style screen.

What you get

  • The full guided learning path: concept lessons, drills, graded tests, and cold-read capstones for every pattern
  • Stop going blank on algorithm problems. The hint ladder walks you to the answer without spoiling it.
  • Actually internalize the trace: dry runs walk state row by row so it clicks before you write a line
  • Know when to reach for each pattern. Every problem names the trigger so you spot it next time.
  • Write and run JS or Python in the browser with real test cases on every problem
  • Access every new problem, unit, and pattern guide as they ship weekly
B
Big_Wolverine_7575Founding Member· unprompted on Reddit

“This app finally made it click. I tried NeetCode, CTCI, and YouTube for years and still couldn't solve Two Sum. The beginner mental models for the patterns are genuinely so helpful. THANK YOU.

Price locks in for life · First 100 only

Founding Member Deal

$29 / year

or $6 / month

LeetCode Premium is $159/year and doesn't teach you anything.

Less than one Udemy course, with guided problems instead of passive lectures.

One extra month of job searching costs more than a full year here.

Your price is locked in for life.

When the 100th founding spot closes, the price goes up for everyone who joins after. Your rate never changes, no matter how much the curriculum grows.

  • The full guided learning path, every unit
  • All 150 problems across 33 patterns, plus every new one
  • New problems and units weekly
  • Cancel anytime

Cancel anytime. Not useful within 7 days? Email for a full refund. Secured by Stripe.