Shohei Yoshida
Shohei Yoshida's Blog

Follow

Shohei Yoshida's Blog

Follow

LeetCode 1971. Find if Path Exists in Graph in F#

Shohei Yoshida's photo
Shohei Yoshida
·Dec 19, 2022·

1 min read

URL

https://leetcode.com/problems/find-if-path-exists-in-graph/description/

Code

https://github.com/syohex/dotnet-study/blob/master/fsharp/leetcode/problems/1971/main.fsx

let validPath (n: int) (edges: (int * int) list) (source: int) (destination: int) : bool =
    let rec validPath' q graph destination visited =
        match q with
        | [] -> false
        | h :: t ->
            if h = destination then
                true
            else
                let visited' = Set.add h visited

                match Map.tryFind h graph with
                | None -> validPath' t graph destination visited'
                | Some(v) ->
                    let q' =
                        v
                        |> List.fold (fun acc node -> if Set.contains node visited' then acc else node :: acc) t

                    validPath' q' graph destination visited'

    let graph =
        edges
        |> List.fold
            (fun acc (s, d) ->
                let acc' =
                    match Map.tryFind s acc with
                    | None -> Map.add s [ d ] acc
                    | Some(v) -> Map.add s (d :: v) acc

                match Map.tryFind d acc' with
                | None -> Map.add d [ s ] acc'
                | Some(v2) -> Map.add d (s :: v2) acc')
            Map.empty

    validPath' [ source ] graph destination Set.empty
 
Share this