题解

binary-tree-postorder-traversal

http://www.nowcoder.com/questionTerminal/32af374b322342b68460e6fd2641dd1b

用栈是正常的思路,但难点在于不要刻意去破坏树的原有结构,所以要用一个prePopedNode来记录上一个被弹栈的节点(由于是后续遍历【压栈顺序是父>右>左】上一个被弹的一定是当前节点的子节点或者左侧节点)

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型ArrayList
     */
    public ArrayList<Integer> postorderTraversal (TreeNode root) {
        // write code here
        ArrayList<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        Stack<TreeNode> stk = new Stack<>();
        stk.push(root);
        // 表示之前没有任何节点被弹过
        TreeNode prePopedNode = null;
        while (!stk.empty()) {
            TreeNode node = stk.peek();

            if (shouldPop(node, prePopedNode)) {
                res.add(node.val);
                stk.pop();
                prePopedNode = node;
                continue;
            }

            if (node.right != null) {
                stk.push(node.right);
            }

            if (node.left != null) {
                stk.push(node.left);
            }
        }

        return res;
    }

    private boolean shouldPop(TreeNode node, TreeNode prePopedNode) {
        // 叶子节点,必弹
        if (node.left == null && node.right == null) {
            return true;
        }

        // 之前弹过
        if (prePopedNode != null) {
            // 之前弹的是其子节点,由于是后续遍历,所以其子节点已经被弹完了,他也必须得弹调
            if (node.left == prePopedNode || node.right == prePopedNode) {
                return true;
            }
            // 之前弹的不是其子节点,所以其子节点还未压栈过,继续压(遍历其子节点)
            else {
                return false;
            }
        }

        return false;
    }
}
全部评论

相关推荐

美团 客服平台 薪资应该是后端算高的了,我们姑且称为nk了,给3w签字费
点赞 评论 收藏
转发
某物流公司 软件开发岗 总包26-30
点赞 评论 收藏
转发
点赞 收藏 评论
分享
正在热议
# 牛客帮帮团来啦!有问必答 #
1151853次浏览 17149人参与
# 通信和硬件还有转码的必要吗 #
11208次浏览 101人参与
# 不去互联网可以去金融科技 #
20436次浏览 256人参与
# 和牛牛一起刷题打卡 #
19005次浏览 1635人参与
# 实习与准备秋招该如何平衡 #
203409次浏览 3627人参与
# 大厂无回复,继续等待还是奔赴小厂 #
4974次浏览 30人参与
# OPPO开奖 #
19213次浏览 267人参与
# 通信硬件薪资爆料 #
265943次浏览 2484人参与
# 国企是理工四大天坑的最好选择吗 #
2227次浏览 34人参与
# 互联网公司评价 #
97709次浏览 1280人参与
# 简历无回复,你会继续海投还是优化再投? #
25037次浏览 354人参与
# 0offer是寒冬太冷还是我太菜 #
454891次浏览 5124人参与
# 国企和大厂硬件兄弟怎么选? #
53909次浏览 1012人参与
# 参加过提前批的机械人,你们还参加秋招么 #
14646次浏览 349人参与
# 硬件人的简历怎么写 #
82289次浏览 852人参与
# 面试被问第一学历差时该怎么回答 #
19398次浏览 213人参与
# 你见过最离谱的招聘要求是什么? #
28147次浏览 248人参与
# 学历对求职的影响 #
161248次浏览 1804人参与
# 你收到了团子的OC了吗 #
538761次浏览 6387人参与
# 你已经投递多少份简历了 #
344255次浏览 4963人参与
# 实习生应该准时下班吗 #
96985次浏览 722人参与
# 听劝,我这个简历该怎么改? #
63525次浏览 622人参与
牛客网
牛客企业服务