LeetCode 404. 左叶子之和

作者:Choi Yang

更新于:7 个月前

字数统计:301 字

阅读时长:1 分钟

题目描述

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

示例：

    3
   / \
  9  20
    /  \
   15   7

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

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/sum-of-left-leaves 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

dfs，求左叶子之和，叶子结点我们比较好判断，而对于左孩子，我们设置一个标记就好了，例如左孩子标记 1，右孩子标记 0，那么当且仅当是叶子节点，并且标记为 1（即左孩子）时，我们进行累加求和。

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var sumOfLeftLeaves = function (root) {
  if (!root) return 0;
  let res = 0;
  let dfs = (cur, root) => {
    // 判断是叶子节点并且是左子树
    if (!root.left && !root.right && cur) {
      res += root.val;
      return;
    }
    // 先遍历左子树在遍历右子树
    // 设置cur为1代表是左子树，为0代表右子树
    root.left && dfs(1, root.left);
    root.right && dfs(0, root.right);
  };
  dfs(0, root);
  return res;
};

学如逆水行舟，不进则退

Contributors

Choi Yang