leetcode 257 二叉树的所有路径

二叉树？寻找路径！

题目
给定一个二叉树，返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。

例子
输入:
—1
–/ – \
2 — 3
-\
–5
输出: [“1->2->5”, “1->3”]
解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3

题目分析

• 起点：根结点
• 终点：叶子结点

1. 从根结点出发

2. 向前走一步到达一个新节点

3. 只要寻找当前结点到叶子结点的路径即可

   ==>重复2,3直到走到叶子结点

   递归！

解题思路

函数	作用
find()	寻找路径的递归函数

过程
当前结点加入路径

• 如果左右子树均为空 —— 当前结点为叶子结点 —— 找到一条路径
• 如果左子树非空 —— 寻找左孩子结点到叶子结点的路径
• 如果右子树非空 —— 寻找右孩子结点到叶子结点的路径

代码如下

void find (TreeNode* now,string path, vector<string> &ans)
 {
    string temp = to_string(now->val);
    if(path=="") path = path + temp;                                 //注意路径的开头不用添加 '->'
    else path = path +'-'+'>' + temp;
    if (!now->left && !now->right) ans.push_back(path);              //路径加入答案
    if(now->left) find(now->left,path,ans);
    if(now->right) find(now->right,path,ans);
 }
class Solution {
public:
    vector<string> binaryTreePaths(TreeNode* root) {
        if(!root) return {};
        string path;
        vector<string> ans;
        find(root, path, ans);
        return ans;
    }
};