【LeetCode】Sum of Left Leaves 左叶子之和

问题

Find the sum of all left leaves in a given binary tree.
给定一棵二叉树,找出其所有的左叶子节点的值的和。

Example:

    3
   / \
  9  20
    /  \
   15   7

There are two left leaves in the binary tree, with values 9 and 15 respectively. Return 24.

思考

这题的关键在于如何找出所有的左叶子节点。而对于一棵二叉树来说,想要找到其所有的左叶子节点,我想到的最直接的方式是遍历这棵二叉树,判断访问到的节点是否为左叶子节点来累加其值,即可得到答案。

解法一

根据我的想法,现在问题转移到了如何遍历获取二叉树的左叶子节点。而遍历一棵二叉树,自然而然的会想到用递归的方式,因为树本身就是一个递归的结构。

二叉树的左叶子节点可能存在于它的左子树和右子树里。从这两部分分别进行讨论求出左叶子节点,再相加即得到所求。

  • 如果二叉树的左子树是一个叶子节点,则不用继续深入下去了,要找的就是它。
  • 如果二叉树的左子树不是一个叶子节点,则递归调用此过程去获取左子树的左叶子节点的值。
  • 而对于二叉树的右子树,则无论其是否是一个叶子节点都不会影响结果,直接递归调用获取右子树的左叶子节点值即可。

按照上述递归的思想,代码如下:

public int sumOfLeftLeaves(TreeNode root) {
    if (root == null) return 0;//递归结束条件
    int left, right;
    if (root.left != null && root.left.left == null 
        && root.left.right == null) {//判断是否为左叶子节点
        left = root.left.val;
    } else {
        left = sumOfLeftLeaves(root.left);
    }
    right = sumOfLeftLeaves(root.right);
    return left + right;
}

解法二

实际上,遍历二叉树的方式实在是太多了,前序遍历、中序遍历、后序遍历、层级遍历等等。遍历二叉树可以通过递归的方式,也可通过迭代的方式。(话说笔试面试中常常会问到树的非递归遍历...=_=)下面通过前序遍历的迭代方式去遍历二叉树,累加左叶子节点的值来解决问题。

代码如下:

public int sumOfLeftLeaves(TreeNode root) {
    TreeNode node = root;
    Stack stack = new Stack<>();
    int sum = 0;
    while (node != null || !stack.isEmpty()) {

        while (node != null) {
            stack.push(node);
            if (node.left != null && node.left.left == null 
                && node.left.right == null) {
                sum += node.left.val;
            }
                node = node.left;
        }

        if (!stack.isEmpty()) {
            node = stack.pop().right;
        }
    }
    return sum;
}

通过一个辅助栈来实现二叉树的非递归前序遍历,在访问到一个节点的时候去判断它是否为左叶子节点即可。

下面再列出层级遍历二叉树的方式去求解的代码:

public int sumOfLeftLeaves(TreeNode root) {
    if (root == null) return 0;
    Queue queue = new ArrayDeque<>();
    int sum = 0;
    queue.offer(root);
    while (!queue.isEmpty()) {
        TreeNode node = queue.poll();
        if (node.left != null && node.left.left == null 
            && node.left.right == null) {
            sum += node.left.val;
        }
        if (node.left != null) queue.offer(node.left);
        if (node.right != null) queue.offer(node.right);
    }
    return sum;
}

思路是一致的,只是层级遍历需要通过一个辅助队列去实现而已。

总结

其实这题挺容易想出解决思路,将求和问题转化为遍历二叉树的问题就行。在这里记录下来,也是为了回顾一下二叉树的几种遍历方式,主要是非递归的几种遍历的写法。

你可能感兴趣的:(【LeetCode】Sum of Left Leaves 左叶子之和)