Leetcode257
链接:力扣 。
题目:
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例1:
输入:root = [1,2,3,null,5]
输出2:["1->2->5","1->3"]
示例2:
输入:root = [1]
输出:["1"]
思路:
很经典的递归加回溯题目。我们同样可以在遍历的代码上进行修改解答此题,根据题目要求,我们要使用递归的前序遍历思路。
用一个vector<string>数组result来记录最终答案,并用一个path数组记录当前已经走过的路径。
用前序遍历的方法。对于当前结点root,如果为空则return,若不为空,判断其是否为叶结点,如果非叶结点则将其加入path,依次递归进入其左子树和右子树,左右子树都递归完后还要从path里面删除该结点。
那么递归的边界在哪里呢?其实存在两个递归边界。
第一个边界就是当前结点为空时,直接return。
第二个边界就是当前访问结点为叶结点时,表示已经走完了一条路径,将该叶结点加入path,然后path里面的结点依次输出并转化为string数组,将其加入result来记录一条结果。然后该叶结点从path里面删除。
参考代码:
/**
* 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> &result) {
if (root == nullptr) {
return;
}
if (root->left == nullptr && root->right == nullptr) {
path.push_back(root->val);
string spath;
for (int i = 0; i < path.size() - 1; i++) {
spath += to_string(path[i]);
spath += "->";
}
spath += to_string(path[path.size() - 1]);
result.push_back(spath);
path.pop_back();
}
else {
path.push_back(root->val);
traversal(root->left, path, result);
traversal(root->right, path, result);
path.pop_back();
}
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<int> path;
vector<string> result;
traversal(root, path, result);
return result;
}
};