Shohei Yoshida
Shohei Yoshida's Blog

Shohei Yoshida's Blog

LeetCode 216. Combination Sum III in F#

Shohei Yoshida's photo
Shohei Yoshida
·May 10, 2022·

1 min read

URL

leetcode.com/problems/combination-sum-iii

Code

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

let combinationSum3 (k: int) (n: int) : int list list =
    let rec combinationSum3' num k n acc : int list list =
        if k = 0 then
            if n = 0 then
                [ (acc |> List.rev) ]
            else
                []
        else
            seq { num..9 }
            |> Seq.filter (fun i -> n - i >= 0)
            |> Seq.map (fun i -> combinationSum3' (i + 1) (k - 1) (n - i) (i :: acc))
            |> Seq.fold (fun acc n -> n @ acc) []

    combinationSum3' 1 k n [] |> List.rev
 
Share this