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

背景介绍:

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

力扣257. 二叉树的所有路径 关于代码的优化
首先看看标准答案代码:

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;
    }
};

总结如下:
1.这种深度优先不要全局变量,然后像回溯一样做,直接用局部变量每个答案都不一样不香吗!
2.to_string将int转换为string
3.1->2 可以考虑为1-> 不用考虑->2(考虑方式改变一下)

上一篇:根据当前的类别查询父级的类别


下一篇:Colin致股东的信,2020经典