给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入:[1,2,3,null,5]
输出:输出: [“1->2->5”, “1->3”]
题解:采用DFS深度优先搜索的办法,利用栈来进行遍历,同时栈中正好保存了路径,每当遇到叶子结点时,就将栈中的现有元素组合成字符串路径。
class Solution {
public List<String> binaryTreePaths(TreeNode root) {
/*
* 用栈来保存路径,同时进行树的遍历,
* 同时要注意的是栈中的每一个结点都有两次搜索机会
* 当一次搜索得到结点的右孩子后,结点不出栈,而是存储在
* 栈中,防止路径顺序被破环,直到从栈中提取的该结点,再弹出栈
*/
Map<TreeNode,Integer>map=new HashMap<>();
List<String>res=new ArrayList<>();
Stack<TreeNode>stack=new Stack<>();
TreeNode node=root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
stack.push(node);
map.put(node,0);
if (node.left == null && node.right == null) {
res.add(getRoute(stack));
}
node = node.left;
}
node = stack.peek();
int temp=map.get(node);
if(temp==1) {
stack.pop();
node=null;
}
else {
map.put(node, 1);
node=node.right;
}
}
return res;
}
private String getRoute(Stack<TreeNode>stack){
StringBuilder route=new StringBuilder();
for(TreeNode x:stack){
route.append(x.val).append("->");
}
int len=route.length();
route.delete(len-2,len);
return route.toString();
}
}