LeetCode题解——257. 二叉树的所有路径

题目相关

题目链接

LeetCode中国，https://leetcode-cn.com/problems/binary-tree-paths/。注意需要登录。

题目描述

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

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

示例

输入:

   1
 /   \
2     3
 \
  5

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

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

题目分析

题意分析

给一个二叉树，从根节点出发分别找出二叉树的左子树和右子树。

样例数据分析

可以看到，输入数据的根节点为 1。我们先遍历左子树，那就是 2。从节点 2 出发，继续遍历左子树，由于左子树不存在，那就遍历右子树，来到了节点 5。由于 5 是叶子节点，因此跟节点的左子树遍历路径为 1->2->5。回到根结点。这次我们遍历右子树，来到了 3。由于 3 是叶子节点，因此跟节点的右子树遍历路径为 1->3。

算法思路

LeetCode 将问题的难度定为简单。我们可以用 DFS 来解决这个问题。

搜索终止条件

当前节点为空。代码如下：

if (NULL==root) {
    return;
}

搜索函数参数

1、一个节点 root，表示当前操作的节点。

2、一个字符串，表示从根节点出发遍历到本节点的路径。

因此 dfs() 函数的原型如下：

void dfs(TreeNode* root, string path);

注意，第二个参数 path 不能使用引用，因为引用会保存原来数据。将导致遍历完左子树后，遍历右子树的时候，左子树的数据还保存着。

开始调用方式

直接从根节点开始调用即可。

string path;
dfs(root, path);

回溯

不需要。

算法步骤

1、从当前节点出发。

2、如果当前节点为空（NULL），则 DFS 函数返回。

3、当前节点不是空，将节点的值加入到遍历过程中。

4、递归遍历当前节点的左子树。

5、递归遍历当前节点的右子树。

AC 参考代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<string> ans;
    vector<string> binaryTreePaths(TreeNode* root) {
        string path;
        dfs(root, path);
        return ans;
    }

    void dfs(TreeNode* root, string &path) {
        //返回条件
        if (NULL==root) {
            return;
        }

        //加入自己
        if (path.size()) {
            path += "->";
        }
        path += to_string(root->val);

        //如果是叶子结点，加入答案
        if (NULL==root->left && NULL==root->right) {
            ans.push_back(path);
        } else {
            //左子树
            dfs(root->left, path);
            //右子树
            dfs(root->right, path);
        }
    }
};