programming-examples/ruby/Algorithms/root_to_leaf_paths.rb

23 lines
589 B
Ruby
Raw Normal View History

2019-11-15 12:59:38 +01:00
# Given a binary tree, return all root-to-leaf paths.
#
# Definition for a binary tree node.
# class TreeNode
# attr_accessor :val, :left, :right
# def initialize(val)
# @val = val
# @left, @right = nil, nil
# end
# end
# @param {TreeNode} root
# @return {String[]}
def root_to_leaf_paths(root, current_path)
paths = []
paths << root.left.nil? ? current_path : root_to_leaf_paths(root.left, current_path + "->#{root.left.val}")
paths << root.right.nil? ? current_path : root_to_leaf_paths(root.right, current_path + "->#{root.right.val}")
paths
end