Shohei Yoshida
Shohei Yoshida's Blog

Shohei Yoshida's Blog

LeetCode 42. Trapping Rain Water in F#

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

1 min read

URL

leetcode.com/problems/trapping-rain-water

Code

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

let trap (height: int list) : int =
    let rec heightMax height (prev: int) acc =
        match height with
        | [] -> acc |> List.tail |> List.rev |> List.tail
        | h :: t ->
            let max = System.Math.Max(h, prev)
            heightMax t max (max :: acc)

    let rec trap' (leftMax: int list) rightMax height acc =
        match leftMax, rightMax, height with
        | [], [], _ -> acc
        | h1 :: t1, h2 :: t2, h3 :: t3 -> trap' t1 t2 t3 (acc + System.Math.Min(h1, h2) - h3)
        | _, _, _ -> failwith "never reach here"

    let leftMax =
        heightMax (List.tail height) (List.head height) [ (List.head height) ]

    let reversed = height |> List.rev

    let rightMax =
        heightMax (List.tail reversed) (List.head reversed) [ (List.head reversed) ]
        |> List.rev

    printf "%A => %A\n" leftMax rightMax
    trap' leftMax rightMax (height |> List.tail) 0
 
Share this