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;
    }
};
发布了34 篇原创文章 · 获赞 0 · 访问量 578

猜你喜欢

转载自blog.csdn.net/Luyoom/article/details/104825650
今日推荐