需要使用set的迭代方法(参照递归方法最好理解)
/**
* 后序遍历
* @param temp
*/
private List<Integer> lrd(TreeNode root) {
LinkedList<Integer> res = new LinkedList<>();
/* 递归方法 */
// if(root== null)
// return;
//
// lrd(root.left);
// lrd(root.right);
// res.addLast(root.val)
/* 非递归方法 */
if(root == null)
return res;
Set<TreeNode> nodeSet = new HashSet<>();
Deque<TreeNode> stack = new LinkedList<>();
stack.addLast(root);
while (!stack.isEmpty()){
TreeNode node = stack.peekLast();
if(nodeSet.contains(node)){
res.addLast(node.val);
stack.removeLast();
} else {
if(node.right != null){
stack.addLast(node.right);
}
if(node.left != null){
stack.addLast(node.left);
}
nodeSet.add(node);
}
}
}
迭代(Leetcode官方题解,不需要set)
后序遍历(LRD)中,根结点是最后添加的那个点,在对二叉树从上到下遍历的过程中,每次访问根节点都把它添加到结果链表的头部(相当于结果链表逆序输出),通过这种方式实现 -------> 实际访问顺序:DRL, 结果输出顺序:LRD。
//二叉树结点的定义
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
LinkedList<TreeNode> stack = new LinkedList<>();
LinkedList<Integer> output = new LinkedList<>();
if (root == null) {
return output;
}
stack.add(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pollLast();
output.addFirst(node.val);
if (node.left != null) {
stack.add(node.left);
}
if (node.right != null) {
stack.add(node.right);
}
}
return output;
}
}
//作者:LeetCode
// 链接:https://leetcode-cn.com/problems/binary-tree-postorder-traversal/solution/er-cha-shu-de-hou-xu-bian-li-by-leetcode/
// 来源:力扣(LeetCode)
// 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。