973. K Closest Points to Origin leetcode

kClosest.go

import "sort"
func kClosest(points [][]int, K int) [][]int {   
    sort.SliceStable(points,func(i,j int)bool{
        point1 := points[i]
        point2 := points[j]
        return distance(point1[0],point1[1]) <  distance(point2[0],point2[1])
    })
    
    return points[:K]
}

func distance(x, y int) int {
	return x*x + y*y
}