LeetCode-257-二叉树的所有路径

Posted 2020-12-11 樱花小猪

LeetCode-257-二叉树的所有路径

题目

给定一个二叉树，返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

示例:

输入:

   1
 /   2     3
   5

输出: ["1->2->5", "1->3"]

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3

思路

这道题就是考察二叉树的遍历，和寻常遍历不同的是需要输出路径，那么只需要在便利的同时加一个形参string；

用string保存路径即可，代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<string> ans;
    void dfs(TreeNode* root, string str) {
        str += to_string(root->val);
        if (!root->left && !root->right) {
            ans.push_back(str);
            return;
        }
        if (root->left) dfs(root->left, str+"->");
        if (root->right) dfs(root->right, str+"->");
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        if (root) dfs(root, "");
        return ans;
    }
};