Sponsored
Sponsored
This approach leverages the simplicity of sorting the list of points based on their distance from the origin. After sorting, the first k points will be the closest ones. The key is to use the squared Euclidean distance to avoid the computational overhead of square root operations.
Time Complexity: O(n log n) due to sorting.
Space Complexity: O(1) since the sorting is done in-place.
1#include <stdio.h>
2#include <stdlib.h>
3
4int compare(const void *a, const void *b) {
5 int *pointA = *(int **)a;
6 int *pointB = *(int **)b;
7 int distA = pointA[0] * pointA[0] + pointA[1] * pointA[1];
8 int distB = pointB[0] * pointB[0] + pointB[1] * pointB[1];
9 return distA - distB;
10}
11
12int** kClosest(int** points, int pointsSize, int* pointsColSize, int k, int* returnSize, int** returnColumnSizes) {
13 qsort(points, pointsSize, sizeof(int*), compare);
14 *returnSize = k;
15 *returnColumnSizes = (int *)malloc(sizeof(int) * k);
16 for (int i = 0; i < k; i++) {
17 (*returnColumnSizes)[i] = 2;
18 }
19 return points;
20}
21This C code uses the qsort function to sort the points array based on the calculated squared distances. It extracts the first k elements and returns them as the closest points.
The Max-Heap approach uses a priority queue to maintain the k closest points seen so far. By using a max-heap, we can efficiently insert new points and potentially evict the farthest point if it is further than any encountered point, leading to a reduced time complexity for finding the k closest points.
Time Complexity: O(n log k) since each insertion/extraction in the heap takes O(log k) time.
Space Complexity: O(k) for the heap storage.
1using System.Collections.Generic;
public class Solution {
public int[][] KClosest(int[][] points, int k) {
PriorityQueue<int[], int> maxHeap = new PriorityQueue<int[], int>(Comparer<int>.Create((a, b) => b - a));
foreach (var point in points) {
int distSq = point[0] * point[0] + point[1] * point[1];
maxHeap.Enqueue(new int[]{point[0], point[1]}, distSq);
if (maxHeap.Count > k) {
maxHeap.Dequeue();
}
}
var result = new List<int[]>();
while (maxHeap.Count > 0) {
result.Add(maxHeap.Dequeue());
}
return result.ToArray();
}
}
This C# solution leverages the PriorityQueue to maintain a max-heap of the k closest points, organized by squared distances, using a custom comparator.