LeetCode-Path Sum

最新推荐文章于 2019-05-12 19:33:21 发布

原创最新推荐文章于 2019-05-12 19:33:21 发布 · 334 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 3 个专栏收录

220 篇文章

订阅专栏

tree

35 篇文章

订阅专栏

DFS

22 篇文章

订阅专栏

本文详细阐述了使用深度优先搜索（DFS）解决二叉树中从根节点到叶子节点的路径求和问题的实现过程。通过分析不同版本的代码，揭示了在编写DFS算法时遇到的挑战，包括basecase的设计、递归函数内部的逻辑判断等，并最终提供了一个简洁高效的解决方案。此外，文章还对比了迭代版本的DFS算法，展示了其在不同场景下的应用优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

典型dfs，但是dfs写的太少了＝＝

base case想了很久，main函数里面怎么处理第一次调用也想了很久。把判断条件搞错了好几次。应该在每次走到这个节点时就判断左右子是否为空。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if ( root == null )
            return false;
        boolean ans = false;
        if ( DFS(root,sum) )
            ans =  true;
        return ans;
    }
    
    public boolean DFS ( TreeNode node, int sum){
        if ( node.left == null && node.right == null && sum == node.val )
            return true;
            
        if ( node.left != null && DFS ( node.left, sum - node.val) == true )
            return true;
        else if ( node.right != null && DFS ( node.right, sum - node.val) == true)
            return true;
        else
            return false;
    }
    
    
    
    
}

下面是别人写的比较清晰

一周多之后又来写了，看了一眼更好的答案：

public class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if ( root == null )
            return false;
        if ( root.val == sum && root.left == null && root.right == null)
            return true;
        return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);
    
    } 
}

iterative:

public class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);

<span style="white-space:pre">	</span>if(root == null)

<span style="white-space:pre">		</span>return false;
     <span style="color:#ff6666;">   while ( !stack.isEmpty()){ </span>
            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;//注意一定先更新val再push
                stack.push(cur.left);
                
            }
            if ( cur.right != null){
                cur.right.val += cur.val;
                stack.push(cur.right);
            }
        }
        return false;
    } 
}