LeetCode 257. 二叉树的所有路径 dfs

地址 https://leetcode-cn.com/problems/binary-tree-paths/

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

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

示例:

输入:

   1
 /   \
2     3
 \
  5

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

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

算法1
DFS遍历

注意由于最后一个元素不加”->”
我是在添加数字的前面才添加”->”
所以要特判第一次加入 if (!path.empty())

C++ 代码

class Solution {
public:
    vector<string> ans;
void Dfs(TreeNode* root, string path) {
    if (root == NULL) {return;}

    if (!path.empty()) { path += "->"; }
    path += to_string(root->val);

    if (root->right == NULL && root->left == NULL) {
        ans.push_back(path);return;
    }

    Dfs(root->right, path);
    Dfs(root->left, path);
}

vector<string> binaryTreePaths(TreeNode* root) {
    string path;
    Dfs(root, path);
    return ans;
}
};