给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入:
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