打印二叉树的路径

最新推荐文章于 2021-04-28 20:59:19 发布

i_am_bird

最新推荐文章于 2021-04-28 20:59:19 发布

阅读量1.2k

点赞数

分类专栏：算法 C++ leetcode

本文链接：https://blog.csdn.net/i_am_bird/article/details/78600345

版权

算法同时被 3 个专栏收录

217 篇文章 0 订阅

订阅专栏

C++

217 篇文章 1 订阅

订阅专栏

leetcode

174 篇文章 2 订阅

订阅专栏

本文介绍了两种方法来打印二叉树的路径：一种是使用递归的方式，从根节点开始，递归遍历左右子树，遇到叶节点时记录路径；另一种是非递归方法，利用两个栈来保存访问路径。提供了相应的代码实现。

摘要由CSDN通过智能技术生成

本题源自leetcode

--------------------------------------------------

思路1 递归：

先访问根节点。然后递归遍历左子树和右子树。遇到叶节点则保存路径。

代码

vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> res;
        if(root==NULL)
            return res;
        findPath(root,res,to_string(root->val));
        return res;
    }
    void findPath(TreeNode* root,vector<string>& res,string s){
        if(!root->left && !root->right){
            res.push_back(s);
            return;
        }
        if(root->left){
            findPath(root->left,res,s+"->"+to_string(root->left->val));
        }
        if(root->right){
            findPath(root->right,res,s+"->"+to_string(root->right->val));
        }
    }

思路2：非递归

用俩个栈来保存访问路径

代码；

vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> res;
        stack<string> str;
        stack<TreeNode*> tree;
        if(root==NULL)
            return res;
        tree.push(root);
        str.push(to_string(root->val));
        while(!tree.empty()){
            TreeNode* p=tree.top();
            tree.pop();
            string s=str.top();
            str.pop();
            if(!p->left && !p->right){
                res.push_back(s);
                continue;
            }
            if(p->left){
                tree.push(p->left);
                str.push(s+"->"+to_string(p->left->val));
            }
            if(p->right){
                tree.push(p->right);
                str.push(s+"->"+to_string(p->right->val));
            }
        }
        return res;
      
    }