LeetCode算法题257:二叉树的所有路径解析

给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:

输入:

   1
 /   \
2     3
 \
  5

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

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

这个题当然还是使用递归,一开始想到了使用辅助函数,但是对于大多数LeetCode题来说是可以有不用辅助函数的方法的,于是在网上搜到了一种方法。思路还是一样的思路,从根开始保存字符串,遇到叶子结点就保存一次,这里转换一下,因为返回的一定是保存字符串的数组,所以可以将所有叶子结点保存到一个数组中,然后递归返回上一层时可以遍历数组所有的元素,然后给其加上当前节点值。
C++源代码:

/**
 * 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> binaryTreePaths(TreeNode* root) {
        vector<string> ret;
        if (root==NULL) return ret;
        if (root->left==NULL && root->right==NULL) 
        {
            ret.push_back(to_string(root->val));
            return ret;
        }
        for (string tmp:binaryTreePaths(root->left))
            ret.push_back(to_string(root->val)+"->"+tmp);
        for (string tmp:binaryTreePaths(root->right))
            ret.push_back(to_string(root->val)+"->"+tmp);
        return ret;
    }
};

python3源代码:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def binaryTreePaths(self, root):
        """
        :type root: TreeNode
        :rtype: List[str]
        """
        l = []
        if root==None: return l
        if root.left==None and root.right==None:
            l.append(str(root.val))
            return l
        for t in self.binaryTreePaths(root.left):
            l.append(str(root.val)+"->"+t)
        for t in self.binaryTreePaths(root.right):
            l.append(str(root.val)+"->"+t)
        return l

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/84145812
今日推荐