题目
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入:
1
/ \
2 3
\
5
输出: ["1->2->5", "1->3"]
解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
思路
这是一个典型的回溯问题。走到一个叶子结点,要查找别的路径时,要先回退到根结点。递归结构就是,当前节点添加到路径中后,先走左子树,后走右子树(相反也可以)。
实现
public List<String> binaryTreePaths(TreeNode root) {
List<String> res = new ArrayList<>();
back(root, new ArrayList<Integer>(), res);
return res;
}
public void back(TreeNode root, List<Integer> list, List<String> res) {
if (root == null) {
return;
}
list.add(root.val);
if (root.left == null && root.right == null) {
StringBuffer sb = new StringBuffer();
for (int num : list) {
if (sb.length() == 0) {
sb.append(num + "");
} else {
sb.append("->").append(num + "");
}
}
res.add(sb.toString());
}
back(root.left, list, res);
back(root.right, list, res);
list.remove(list.size() - 1);
}