Heaps and Priority Queues: An Interview Playbook

Reach for a heap when the problem keeps asking for the best of a changing set: the smallest, the largest, the closest, the soonest. Not once. Over and over, as items come and go.

The two interview families

1) Top K / threshold maintenance

Keep the heap size bounded. Examples: top K frequent elements, K closest points, median from a stream (often two heaps).

2) Merge / schedule by time

Push candidates into a min-heap ordered by next event time. Examples: merge K sorted lists, task scheduler variants, Dijkstra-lite situations on small graphs.

Complexity checklist

• Push and pop are typically O(log n) per operation.
• Building a heap from an array can be O(n) if you heapify in place (a common talking point).
• Peek is O(1) when the heap already has elements.

Implementation traps

• Python heapq is a min-heap. For max-heap tricks, negate values or store tuples carefully.
• Tie-breaking means if the problem needs stable-ish ordering, your tuple keys matter.
• Do not heapify objects unless you know your comparator story.

Pattern recognition drill

When you read a problem, ask yourself structured questions.

1. Is there a repeated "pick the best next" step?
2. Can I bound the heap size to K or to the frontier of a graph search?

If both answers lean yes, sketch the heap before you reach for sorting everything.

Practice pairing means solving one heap problem timed, then redoing it the next day from a blank file. The second pass should feel boring, and that is the point.

How to practice this skill

Treat this topic as a loop rather than a fact to memorize. First, name the pattern in plain English. Then write the smallest version that proves you understand the invariant. Only after that should you chase speed. That order matters because most interview mistakes are not typing mistakes; they are recognition mistakes. You reach for the right data structure too late, or you optimize before you have named what must stay true.

For heaps practice, the useful repetition is spaced. Solve one problem slowly, rewrite the solution from memory the next day, then do a timed variant after the idea feels boring. The second pass is where the pattern moves from recognition to recall. The timed pass is where it becomes usable under pressure.

Implementation checklist

• Restate the input and output before coding.
• Write down the invariant or decision rule in one sentence.
• Test the smallest case, the largest obvious case, and the case that breaks the naive approach.
• Compare the final complexity to the constraint that actually matters.

The checklist is intentionally short. A long checklist becomes another thing to memorize. A short one becomes a rhythm you can run in a blank editor, in a live interview, or in an AlgoArena battle.

Edge cases worth drilling

Most missed solutions come from one of three places: empty input, duplicated values, or a boundary that looks harmless until the loop reaches it. When you review a solution, do not only ask whether it passed. Ask which boundary made the implementation honest.

If a solution uses indexes, trace the first and last iteration. If it uses a map or set, trace the duplicate case. If it uses recursion, name the base case and the state that gets smaller. These small reviews are faster than solving a new problem and often teach more.

Review drill

Before you move on, turn the idea into a small review drill. Pick one representative problem and write three things before you code: the pattern name, the data structure you expect to use, and the edge case that would make the naive version fail. After you solve it, rewrite the explanation in two sentences without looking at the code.

That last step is the difference between recognizing a solution and owning it. If you can explain why the approach works, you can usually rebuild it under pressure. If you only remember the final code shape, the next variant will feel like a new problem.

On AlgoArena, pair this with a timed rep only after the slow explanation is clean. Speed is useful when it compresses a skill you already understand. It is noisy when it hides the fact that the invariant was never clear.

One-session exercise

End with one mixed rep. Choose a problem where the pattern is present but not named in the title. Spend five minutes identifying the signal, ten minutes coding, and five minutes writing a post-solve note about what made the pattern recognizable. If you cannot name the signal before coding, the next best rep is not a harder problem. It is another example of the same pattern with different surface details.

That small debrief is what turns a guide into skill. You are training yourself to notice the shape before the solution is obvious.

Repeat the same exercise once with notes open and once with notes closed. The gap between those two attempts tells you whether the idea is still borrowed or has become part of your own toolkit.

If the closed-notes pass fails, keep the problem but shrink the scope: trace one loop, one recursive call, or one state transition until the reason is obvious.