Extracting Leaf paths from n-ary tree in F#

后端 未结 2 1225
傲寒
傲寒 2021-01-18 10:59

Inspired by this question, I wanted to try my hand at the latest ponder this challenge, using F#

My approach is probably completely off course, but in the course of

2条回答
  •  不知归路
    2021-01-18 11:46

    regarding your question about list traversal - you can start by writing a function that returns lists that represent the path - that's I think easier and it will be later easy to turn it into a function that returns a number.

    This one takes a list as the first argument (path so far) and a tree and returns a list> type - that is all the possible paths from the current branch.

    let rec visitor lst tree = 
      match tree with
      | Branch(n, sub) -> List.collect (visitor (n::lst)) sub
      | Leaf(n) -> [List.rev (n::lst)]
    
    // For example...
    > let tr = Branch(1, [Leaf(3); Branch(2, [Leaf(4); Leaf(5)] )]);;
    > visitor [] tr;;
    val it : int list list = [[1; 3]; [1; 2; 4]; [1; 2; 5]]
    

    In the 'Leaf' case, we simply add the current number to the list and return the result as a list containing single list (we have to reverse it first, because we were adding numbers to the beginning). In the 'Branch' case, we add 'n' to the list and recursively call the visitor to process all the sub-nodes of the current branch. This returns a bunch of lists and we use 'map_concat' to turn them into a single list that contains all posble paths from the current branch.

    Now, you can rewrite this to return a list of integers:

    let rec visitor2 lst tree = 
      match tree with
      | Branch(n, sub) -> List.collect (visitor2 (lst * 10 + n)) sub
      | Leaf(n) -> [lst * 10 + n]
    
    // For example...  
    > visitor2 0 tr;;
    val it : int list = [13; 124; 125]  
    

    Instead of concatenating lists, we now calculate the number.

提交回复
热议问题