LeetCode 257. 二叉树的所有路径【C++】

题目

给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [1,2,3,null,5]
输出：[“1->2->5”,“1->3”] 示例 2：

示例 2：

输入：root = [1]
输出：[“1”]

提示：

• 树中节点的数目在范围 [1, 100] 内
• -100 <= Node.val <= 100

代码

1. 使用递归 前序遍历

class Solution {
public:
void allPath(TreeNode* node,vector<int>& path,vector<string>& res){
    path.push_back(node->val);
    
    //找到叶子节点时，终止
    if(node->left==nullptr && node->right==nullptr){
    
    //使其符合 输出格式
        string patht;
        for(int i=0;i<path.size()-1;i++){ 
            patht+=to_string(path[i]);
            patht+="->";
        }
        patht+=to_string(path[path.size()-1]);
        res.push_back(patht);
        return;

    } 

    if(node->left){
        allPath(node->left,path,res);
        path.pop_back();    //回溯
    }
    if(node->right){
        allPath(node->right,path,res);
        path.pop_back();    //回溯
    }
}
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<int> path;
        vector<string> res;
        if(root==nullptr) return res;
        allPath(root,path,res);
        return res;

    }
};

2. 迭代法 前序遍历

class Solution {
public:

    vector<string> binaryTreePaths(TreeNode* root) {
        stack<string> paths;	//路径
        vector<string> res;
        stack<TreeNode*> troot;	//树的遍历节点
        if(root==nullptr) return res;
        troot.push(root);
        paths.push(to_string(root->val));
        while(!troot.empty()){
            TreeNode* node=troot.top();	//中
            troot.pop();
            string path=paths.top();	//去除该节点对应的路径
            paths.pop();
            if(node->left==nullptr && node->right==nullptr) res.push_back(path);
            if(node->left){				//左
                troot.push(node->left);
                paths.push(path+"->"+to_string(node->left->val));
            }
            if(node->right){
                troot.push(node->right);	//右
                paths.push(path+"->"+to_string(node->right->val));
            }
        }
        return res;

    }
};

---

来源：力扣（LeetCode）

链接：https://leetcode.cn/problems/binary-tree-paths/description/