Leetcode 404 左葉子之和
阿新 • • 發佈:2020-12-04
Leetcode 404 左葉子之和
資料結構定義:
計算給定二叉樹的所有左葉子之和。
示例:
3
/ \
9 20
/ \
15 7
在這個二叉樹中,有兩個左葉子,分別是 9 和 15,所以返回 24
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
遞迴寫法:
class Solution { int sum = 0; public int sumOfLeftLeaves(TreeNode root) { postOrderTraversal(root); return sum; } private void postOrderTraversal(TreeNode root){ if(root == null){ return; } if(root.left != null && root.left.left == null && root.left.right == null){ sum += root.left.val; } postOrderTraversal(root.left); postOrderTraversal(root.right); } }
另一種遞迴方式:
class Solution { public int sumOfLeftLeaves(TreeNode root) { if (root == null) return 0; return sumOfLeftLeaves(root.left) + sumOfLeftLeaves(root.right) + (isLeft(root) ? root.left.val : 0); } private boolean isLeft(TreeNode root) { return (root.left != null && root.left.left == null && root.left.right == null); } }
廣度優先遍歷方式
class Solution {
public int sumOfLeftLeaves(TreeNode root) {
if(root == null){
return 0;
}
int sum = 0;
Queue<TreeNode> queue =new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode node = queue.poll();
if(node.left != null){
if(isLeft(node.left))
sum += node.left.val;
else
queue.offer(node.left);
}
if(node.right != null)
queue.offer(node.right);
}
return sum;
}
private boolean isLeft(TreeNode root){
return root.left == null && root.right == null;
}
}