非递归方法遍历树(前序遍历、中序遍历及后续遍历)

public List<Integer> preorderTraversal(TreeNode root) {
     List<Integer> res = new ArrayList<>();
     if (root == null) return res;
     LinkedList<TreeNode> stack = new LinkedList<>();
     stack.push(root);
     while (!stack.isEmpty()) {
         TreeNode node = stack.pop();
         res.add(node.val);
         if (node.right != null) stack.push(node.right);
         if (node.left != null) stack.push(node.left);
     }
     return res;
 }

 public List<Integer> inorderTraversal(TreeNode root) {
     List<Integer> res = new ArrayList<>();
     if (root == null) return res;
     LinkedList<TreeNode> stack = new LinkedList<>();
     TreeNode currNode = root;

     while (currNode != null) {
         stack.push(currNode);
         currNode = currNode.left;
     }

     while (!stack.isEmpty()) {
         currNode = stack.pop();
         res.add(currNode.val);
         currNode = currNode.right;
         while (currNode != null) {
             stack.push(currNode);
             currNode = currNode.left;
         }
     }

     return res;
 }

 public List<Integer> postorderTraversal(TreeNode root) {
     List<Integer> res = new LinkedList<>();
     if (root == null) return res;
     LinkedList<TreeNode> nodeStack = new LinkedList<>();
     Set<TreeNode> nodeSet = new HashSet<>();

     TreeNode currNode = root;
     while (currNode != null) {
         nodeStack.push(currNode);
         currNode = currNode.left;
     }

     while (!nodeStack.isEmpty()) {
         currNode = nodeStack.peek();
         if (currNode.right != null && !nodeSet.contains(currNode.right)) {
             currNode = currNode.right;
             while (currNode != null) {
                 nodeStack.push(currNode);
                 currNode = currNode.left;
             }
         } else {
             currNode = nodeStack.pop();
             res.add(currNode.val);
             nodeSet.add(currNode);
         }
     }

     return res;
 }

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值