Greedymedium45 minAI/ML Engineer

Batching Greedy Load Rebalancer

AI/ML Engineer signal: greedy proof + sorting in a model-serving batcher context. This is a ProdMatch-owned ai ml engineer drill, framed as a March 2026 Paytm Inference Platform simulation, not a copied platform question.

Company context

Paytm · Inference Platform

Freshness

March 2026

Product surface

model-serving batcher

ProdMatch interview simulation based on product-team patterns; not a claim of a real company question.

Question

Rebalance requests across shards for model-serving batcher. Each move has a cost equal to distance between shards. Compute minimum cost to make shard loads differ by at most one.

Input

Array of shard loads.

Output

Minimum movement cost.

Constraints

1 <= shards <= 200000
Total load <= 1e12

Concepts

vector search
RAG retrieval
recommendation graphs
greedy proof
sorting
load balancing

loads [0,3,0] -> move one unit left and one right, cost=2

Approach

Try framing your own approach first. The 30 seconds you think before peeking is where learning happens.

Clean Solution

Reveal the approach first.

How well did you understand?

Your rating tunes when this problem shows up again.

Common Mistakes

A greedy idea without a proof may fail hidden cases.
Do not mutate sorted keys in a way that breaks ordering.

Next Similar Problems

Vector Greedy Load Rebalancermedium GraphRAG Greedy Load Rebalancermedium Recommend Greedy Load Rebalancerhard