力扣257. 二叉树的所有路径关于代码的优化

最新推荐文章于 2024-11-10 14:06:08 发布

奋斗吧！骚年！

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量478

点赞数

分类专栏：算法杂记文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_46470984/article/details/122026108

版权

算法杂记专栏收录该内容

6 篇文章 0 订阅

订阅专栏

背景介绍：

题目虽然做出来了，但是总是跟标准答案差一点优化，所以写下此博客来做笔记。

在这里插入图片描述
首先看看标准答案代码：

class Solution {
public:
    void dfs(TreeNode* root,string path,vector<string> &paths){
        path+=to_string(root->val);
        if(root->left==nullptr&&root->right==nullptr){
            paths.push_back(path);
            return;
        }else{
            path+="->";
            if(root->left!=nullptr)dfs(root->left,path,paths);
            if(root->right!=nullptr)dfs(root->right,path,paths);
        }
        
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> paths;
        dfs(root,"",paths);
        return paths;
    }
};

再看看我的代码

class Solution {
public:
    vector<string> res;
    vector<int> mid;
    void dfs(TreeNode *root){
        mid.push_back(root->val);
        if(root->left==nullptr&&root->right==nullptr){
            string s="";
            for(int i=0;i<mid.size();i++){
                if(i!=0)s+="->";
                s+=to_string(mid[i]);
                
            }
            res.push_back(s);
            mid.pop_back();
            return;
        }
        if(root->left!=nullptr)dfs(root->left);
        if(root->right!=nullptr)dfs(root->right);
        mid.pop_back();
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        dfs(root);
        return res;
    }
};