力扣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(考虑方式改变一下)
标签:paths,right,string,nullptr,dfs,力扣,二叉树,root,257 来源: https://blog.csdn.net/qq_46470984/article/details/122026108