101 Logo
onenoughtone

Problem Statement

Proximity Point Finder

You're developing a mapping application that needs to find the closest points of interest to the user's current location.

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).

The distance between two points on a plane is the Euclidean distance: √((x1 - x2)² + (y1 - y2)²).

You may return the answer in any order. The answer is guaranteed to be unique (except for the order that it is in).

Examples

Example 1:

Input: points = [[1,3],[-2,2]], k = 1
Output: [[-2,2]]
Explanation: The distance between (1, 3) and the origin is sqrt(10). The distance between (-2, 2) and the origin is sqrt(8). Since sqrt(8) < sqrt(10), (-2, 2) is closer to the origin. We only want the closest k = 1 points from the origin, so the answer is just [[-2,2]].

Example 2:

Input: points = [[3,3],[5,-1],[-2,4]], k = 2
Output: [[3,3],[-2,4]]
Explanation: The answer [[-2,4],[3,3]] would also be accepted.

Constraints

  • 1 <= k <= points.length <= 10^4
  • -10^4 < points[i][0], points[i][1] < 10^4

Problem Breakdown

To solve this problem, we need to:

  1. The Euclidean distance from a point (x, y) to the origin (0, 0) is sqrt(x² + y²)
  2. Since we only care about comparing distances, we can use x² + y² instead of sqrt(x² + y²) to avoid floating-point calculations
  3. We need to find the k points with the smallest distances to the origin
  4. A max heap of size k can efficiently maintain the k closest points
  5. Alternatively, we can sort all points by their distances and return the first k points
  6. The QuickSelect algorithm can find the k closest points in O(n) average time
ProblemSolutionCode
101 Logo
onenoughtone