swift 算法 简单50.二叉树的所有路径

给定一个二叉树，返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

示例:

输入:

   1
 /   \
2     3
 \
  5

输出: ["1->2->5", "1->3"]

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3

解法：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public var val: Int
 *     public var left: TreeNode?
 *     public var right: TreeNode?
 *     public init(_ val: Int) {
 *         self.val = val
 *         self.left = nil
 *         self.right = nil
 *     }
 * }
 */
class Solution {
    func binaryTreePaths(_ root: TreeNode?) -> [String] {
     var array = [String]()
        let str:String = ""
        self.getTreeNode(&array, root,str)
        return array
    }
    func getTreeNode(_ array: inout [String] ,_ root: TreeNode?,_ str:String) {
        var str = str
        guard root != nil else {
            return
        }
        
        if root?.left == nil && root?.right == nil {
            str.append(String(root!.val))
            array.append(str)
        }else{
            getTreeNode(&array, root?.left,str+String(root!.val)+"->")
  
            getTreeNode(&array, root?.right,str+String(root!.val)+"->")
        }
        
    }
}