LeetCode-Binary Tree Postorder Traversal

来源:互联网 发布:c语言平方怎么写 编辑:程序博客网 时间:2024/06/05 11:33

题目不用多说,就是二叉树的后序遍历。如果是递归进行遍历,相信大家是都会的。这里说下用栈进行遍历的大体思路及代码:

思路

与前序遍历不同,如果用栈实现后序遍历,当遍历完某个子树后,需要判断之前遍历的是左子树还是右子树。如果是左子树,则要再遍历右子树。如果刚遍历的是右子树,则要访问根节点,并出栈。

代码

public class Solution {    public List<Integer> postorderTraversal(TreeNode root) {        List<Integer> result = new ArrayList<Integer>();        if(root == null){            return result;        }        Deque<TreeNode> stack = new ArrayDeque<TreeNode>();        TreeNode prev = null;        while(true){            while(root != null){                stack.push(root);                root = root.left;            }            if(!stack.isEmpty()){                TreeNode top = stack.peek();                if(top.right == null || (prev != null && prev == top.right)){                    prev = stack.pop();                    result.add(prev.val);                }else{                    root = top.right;                }            }else{                break;            }        }        return result;    }}
0 0
原创粉丝点击