LeetCode: 404.左叶子节点

Posted 2020-11-23 tiana_Z

　　计算给定二叉树的所有左叶子之和。

　　示例：

　　3

　　/

　　9 20

　　/

　　15 7

　　在这个二叉树中，有两个左叶子，分别是 9 和 15，所以返回 24

　　解析

　　我们需要找到这样的节点

　　属于叶子节点

　　属于父节点的左子节点

　　方法一：用栈，dfs遍历，用全局变量res作为累积和。遍历的过程中传递该节点是否是左子节点。同时判断左右子节点是否为None，则可以知道是不是左叶子节点。

　　class Solution:

　　def sumOfLeftLeaves(self, root: TreeNode) -> int:

　　stack = []

　　res = 0

　　if not root:

　　return res

　　stack.append((root, 0))

　　while len(stack) != 0:

　　p, flag = stack.pop()

　　if flag == 1:

　　if p.left == p.right == None:

　　res += p.val

　　if p.right:

　　stack.append((p.right, 0))

　　if p.left:

　　stack.append((p.left, 1))

　　return res

　　方法二，用递归遍历。遍历到左叶子节点则加上左叶子节点的值。

　　class Solution:

　　def sumOfLeftLeaves(self, root: TreeNode) -> int:

　　self.res = 0

　　def walk(p):

　　if p:

　　if p.left:

　　if p.left.left == p.left.right == None:

　　self.res += p.left.val

　　walk(p.left)

　　if p.right:

　　walk(p.right)

　　walk(root)

　　return self.res

　　方法三， 仍是递归，没有使用全局变量。而是使用在递归函数内部累积的方式(即有返回值)。遍历到左叶子节点，则返回值就在此基础上加上右节点的遍历。

　　class Solution:郑州哪家人流医院好 http://www.gz020zj.com/

　　def sumOfLeftLeaves(self, root: TreeNode) -> int:

　　if root == None:

　　return 0

　　res = 0

　　if root.left:

　　if root.left.left == root.left.right == None:

　　res += root.left.val

　　return res + self.sumOfLeftLeaves(root.left) + self.sumOfLeftLeaves(root.right)

　　方法四，在递归的过程中，用一个形参记录该节点是否为左孩子点。和用stack遍历类似。

　　class Solution:

　　def sumOfLeftLeaves(self, root: TreeNode) -> int:

　　def cal(p, dir):

　　if not p:

　　return 0

　　if p.left == p.right == None:

　　if dir == 1:

　　return p.val

　　else:

　　pass

　　return cal(p.left, 1) + cal(p.right, 0)

　　return cal(root, 0)

　　方法五，当然还能用bfs遍历，遍历到左叶子节点就加上去。