LeetCode: 404.左叶子节点

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

  示例:

  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遍历,遍历到左叶子节点就加上去。

posted @ 2019-12-17 16:29  tiana_Z  阅读(136)  评论(0编辑  收藏  举报