LeetCode相同的树(Python)——深度、广度优先遍历

题目

LeetCode相同的树(Python)——深度、广度优先遍历

解题思路

关于二叉树这个数据结构,考察比较多的就是深度优先遍历和广度优先遍历,其中深度优先遍历还分前、中、后序遍历。先实现一下简单的深度优先遍历:

深度优先遍历中的前序遍历:
LeetCode相同的树(Python)——深度、广度优先遍历
LeetCode相同的树(Python)——深度、广度优先遍历
深度优先遍历的中序遍历:
LeetCode相同的树(Python)——深度、广度优先遍历
深度优先遍历中的后序遍历:
LeetCode相同的树(Python)——深度、广度优先遍历
执行结果为:
LeetCode相同的树(Python)——深度、广度优先遍历
接下来实现难一点的广度优先遍历,这里也是借助了官方题解中的思路,模拟队列这样的数据结构来存放节点,每次弹出队首的节点进行比较,同时把这个节点的左右子节点加入队尾(如果有的话),代码如下:
LeetCode相同的树(Python)——深度、广度优先遍历
LeetCode相同的树(Python)——深度、广度优先遍历
在官方的实现中,直接调用了collections类中的双向列表deque
(什么是deque?当我们使用list存储数据时,按索引访问元素很快,但是插入和删除元素很慢,之前也提到过几次,数据量大的时候,因为需要“搬动”数据,导致插入和删除效率很低。deque是为了高效实现插入和删除操作的双向列表,适合用于队列和栈):
LeetCode相同的树(Python)——深度、广度优先遍历

在我的实现中,因为只用到了队列先进先出这一个特性,且append恰好是默认在列表最后的地方加入数据,不需要“搬动”其他数据,所以还是直接使用了list,但是为了模拟队列弹出队首元素的同时避免使用del,我使用 i += 1 模拟删除了队首元素。

执行结果为:
LeetCode相同的树(Python)——深度、广度优先遍历