We're live on Product Hunt today! Use code PRODUCTHUNT26 for 20% off. Upvote us ↑
DSA Trainer
← All problems
PremiumHeap / Priority QueueMedium

K Closest Points to Origin

Asked at:AmazonGoogleMeta

Given an array of points on a 2D plane, return the k points closest to the origin (0, 0). Distance is Euclidean. The answer can be in any order.

This is a premium problem.

Unlock K Closest Points to Origin with both full courses, DSA and System Design, and every other premium problem.

Problem

Given an array of points where points[i] = [xi, yi] represents a point on the X-Y plane and an integer k, return the k closest points to the origin (0, 0).

Input

An array of `points` where `points[i] = [xi, yi]`, and an integer `k`.

Output

The k points closest to the origin.

Examples

Input: points = [[1,3],[-2,2]], k = 1

Output: [[-2,2]]

Distance² from origin: [1,3] → 10, [-2,2] → 8. Closest is [-2,2].

Input: points = [[3,3],[5,-1],[-2,4]], k = 2

Output: [[3,3],[-2,4]]

The brute-force approach

Sort all points by distance from origin and return the first k.

points.sort(key=lambda p: p[0]**2 + p[1]**2)
return points[:k]

O(n log n) — sorting the full array even if k is much smaller than n.

Time: O(n log n)Space: O(1)

Spotting the pattern

This is a Heap / Priority Queue problem. The key question to ask yourself:

To keep only the k closest points seen so far, you need to quickly find and remove the farthest. Which data structure finds the maximum in O(1) and removes it in O(log k)?

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 Heap / Priority Queue walkthrough for K Closest Points to Origin: the progressive hint ladder, a row-by-row dry run, the optimized code, and an in-browser runner, plus both full courses, DSA and System Design, 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 DSA course: concept lessons, drills, graded tests, and cold-read capstones for every pattern
  • The full System Design course: 20 building-block units for the design round, included at no extra cost
  • 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.

Full access · both courses

Full access

$49 / year

or $9 / month

Now includes a C++ runner alongside JavaScript and Python.

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.

  • Both courses, every unit: DSA + System Design
  • 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.