102.二叉树的层序遍历
层序遍历其实和之前的前中后序类似,但是它的不同点在于,它是广度优先搜索。也就是考虑完一个点后,先考虑它四周扩散开来的点,然后以此类推。那么这里使用队列是很方便的。如果单纯得到一个一维数组的话,其实本题思路和之前的完全一样,只要每次考虑一个点,然后把他的左右加到队列里面,自然是有顺序的。但是本题难点在于它的数组有层数之分,因此我们要考虑每一个节点是哪一层的。简单的想法就是,每一个保存的节点,都加一个层数标记,然后后续分开考虑,不过这样较为麻烦。更好的方法是直接一层一层的处理,那么如何确保每次处理的是同一层的呢,此时需要规定好每次操作的次数,要知道该层有多少个节点。此时可以记录下每层长度,记录方法就是处理完上一层后,queue中只含有当前层的节点,那么提前记录长度就很方便。题解如下:
101. 对称二叉树
本题难点在于对称的判断,它是要求左边的右边和右边的左边相等才行。那么我们判断应该是root.left.right == root.right.left类似的条件。然后对于每一次的判断,可以有以下四种情况,两个值相等,那么此时我们并不能确定这个就是对称了,只有两个值都是None的时候才是对称(第二种情况),因此此时只能继续往下判断。而第三种情况是两者值不同,那么可以直接返回False了。最后一种是其中一个为None,另一个不是,此类情况也可以说明不对称,返回False。题解如下:
226.翻转二叉树
本题相对来说较为简单,只要把根节点左右互换即可。而对于每个子节点,都进行类似的操作。那么就可以简单使用递归完成。题解如下:
class Solution:
def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
if not root:
return
root.left, root.right = self.invertTree(root.right), self.invertTree(root.left)
return root