Time: 62 ms (49.1%), Space: 64.4 MB (78.61%) - LeetHub

Author: hajin-kim

0973-k-closest-points-to-origin/0973-k-closest-points-to-origin.kt

@@ -1,15 +1,8 @@
 class Solution {
     fun kClosest(points: Array<IntArray>, k: Int): Array<IntArray> {
-        val distances = points
-            .map { (x, y) ->
-                x * x + y * y
-            }
-
         return points
-            .withIndex()
-            .sortedBy { (index, _) -> distances[index] }
+            .sortedBy { (x, y) -> x * x + y * y }
             .take(k)
-            .map { it.value }
             .toTypedArray()
     }
 }