题解 |【通俗易懂】 #二叉树的中序遍历#颜色标记这个好

已默认你已经读懂题意

在树的深度优先遍历里面(包括前序、中序、后序这三种遍历),做二叉树的中序遍历其实递归方法是最为直观易懂,但是考虑到效率的问题,我的建议不使用递归。

另外就是栈迭代的方法虽然确实提高了效率,但嵌套循环太多导致非常的烧脑,不易理解,容易造成“一看就懂,一写就废”的尴尬,另外对于其他不同的遍历顺序(如前序、中序、后序),循环结构差异很大,不能做到代码的复用。

所以,这个咋办啊

做过遗传学大题没有?

还是熟悉的味道

母患,子必患...

都扯到外婆家去了,可以用一种“颜色标记法”(瞎起的名字……),这种方法的好处兼具栈迭代方法的高效,又能像递归方法一样简洁容易懂,更重要的是,这种方法对于前序、中序、或者后序遍历,都能够写出完全一致的代码,符合真香定律。

我的解题思路如下

用颜色标记节点的状态,新的结点标记成白色,已经访问过的结点标记成灰色;那么就有两种情况,如果遇到的是白色结点,下一步把它变成灰色结点,然后把该结点的右边的子节点、自身、左边的子节点依次入栈;如果你遇到的是灰色结点,则把节点的值进行输出。

使用这种方法实现的中序遍历代码如下

//python这么来写
class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        WHITE, GRAY = 0, 1
        res = []
        stack = [(WHITE, root)]
        while stack:
            color, node = stack.pop()
            if node is None: continue
            if color == WHITE:
                stack.append((WHITE, node.right))
                stack.append((GRAY, node))
                stack.append((WHITE, node.left))
             else:
                res.append(node.val)
             return res
全部评论

相关推荐

01-17 08:34
门头沟学院 Java
想找对象的单身狗在努力存钱:这工资不低了,再高点人家要招博士硕士的
点赞 评论 收藏
分享
Just路人甲:MUC 是什么🤔,32不是 MCU 吗
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务