1. 程式人生 > >112 路徑總和

112 路徑總和

給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例: 
給定如下二叉樹,以及目標和 sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        boolean flag = false;
        if (root == null
){ return false; } Stack<TreeNode> stack = new Stack<TreeNode>(); stack.push(root); while (!stack.empty()){ TreeNode cur = stack.pop(); if (cur.val == sum && cur.left == null && cur.right == null){
return true; } if (cur.left != null){ cur.left.val += cur.val; stack.push(cur.left); } if (cur.right != null){ cur.right.val += cur.val; stack.push(cur.right); } } return flag; } }