Leetcode 257. 二叉树的所有路径

Leetcode 257. 二叉树的所有路径

题目

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

叶子节点 是指没有子节点的节点。

思路

• 创建一个vectorpath，保存先序遍历的路径节点，然后创建一个vectorresult,保存路径
• 先序遍历，首先保存中间节点的节点值，然后判断当前节点是不是叶子节点，如果是叶子节点，说明需要进行路径处理逻辑，循环遍历path,将每一个节点存入字符串中，然后再将字符串存入result
• 左子树不为空，递归遍历左子树，右子树不为空，递归遍历右子树

代码

/**
 * 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* cur,vector<int>& path,vector<string>&result)
    {
        // 记录节点
        path.push_back(cur->val);
        // 遇到叶子节点
        if(cur->left == NULL && cur->right == NULL)
        {
            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);
            return;
        }

        if(cur->left !=NULL)
        {
            traversal(cur->left,path,result);
            path.pop_back();// 回溯
        }

        if(cur->right != NULL)
        {
            traversal(cur->right,path,result);
            path.pop_back();// 回溯
        }
    }

    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;// 保存路径
        vector<int> path;// 保存节点

        if(root == NULL)
        {
            return result;
        }

        traversal(root,path,result);
        return result;
    }
};