257 二叉树的所有路径
题目
返回二叉树从根节点到叶子节点的所有路径。
思路
按照前序遍历的顺序,访问每个叶子节点,不过需要注意的是在递推过程中,保存当前叶子节点的路径后,需要回退一个节点才能进入另一个路径。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void traversal(TreeNode * root,vector<int>& path,vector<string>& res) {
path.push_back(root->val);
if(!root->left && !root->right) {
string temp;
for(int i = 0;i < path.size() - 1;i++) {
temp += to_string(path[i]);
temp += "->";
}
temp += to_string(path[path.size() - 1]);
res.push_back(temp);
return ;
}
if(root->left) {
traversal(root->left,path,res);
path.pop_back(); //回退到父节点才能进入右孩子
}
if(root->right) {
traversal(root->right,path,res);
path.pop_back();
}
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<int> path;
vector<string> res;
if(!root) return res;
traversal(root,path,res);
return res;
}
};