leetcode257(二叉树的所有路径:二叉树搜索)

tech2025-03-30  11

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

示例: 输入:[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(); } }
最新回复(0)