Shohei Yoshida
Shohei Yoshida's Blog

Shohei Yoshida's Blog

LeetCode 658. Find K Closest Elements in F#

Shohei Yoshida's photo
Shohei Yoshida
·Sep 29, 2022·

1 min read

URL

leetcode.com/problems/find-k-closest-elements

Code

github.com/syohex/dotnet-study/blob/master/..

let findClosestElements (arr: int list) (k: int) (x: int) : int list =
    arr
    |> List.sortWith (fun a b ->
        let aAbs = System.Math.Abs(a - x)
        let bAbs = System.Math.Abs(b - x)

        if aAbs = bAbs then
            compare a b
        else
            compare aAbs bAbs)
    |> List.take k
    |> List.sort

// [1;2;3;4]
findClosestElements [ 1; 2; 3; 4; 5 ] 4 3

// [1;2;3;4]
findClosestElements [ 1; 2; 3; 4; 5 ] 4 -1
 
Share this