欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode:二叉树的后序遍历【145】

程序员文章站 2022-03-03 10:54:05
...

LeetCode:二叉树的后序遍历【145】

题目描述

给定一个二叉树,返回它的 后序 遍历。

示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [3,2,1]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

题目分析

  这道题难度和N叉树的后序遍历是等同的,但是标注为困难。

  首先我们都知道,栈顶元素一般都是根元素,弹出根元素,加入根元素的左节点、右节点后,位于栈顶的是根元素的最右子节点,栈低的是根元素的最左子节点

  如果我们按照这个顺序打印的话,输出的是[1,2,3]。如果我们把这个结果反转后就是[3,2,1]就是结果。

Java题解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
          Stack<TreeNode> stack = new Stack<>();
        List<Integer> ans = new ArrayList<>();
        if(root==null)
            return ans;
        stack.push(root);
        while (!stack.isEmpty())
        {
            TreeNode node = stack.pop();
            if(node.left!=null)
                stack.add(node.left);
            if(node.right!=null)
                stack.add(node.right);
            ans.add(node.val);
        }
        Collections.reverse(ans);
        return ans ;
    }
}