JZ82 二叉树中和为某一值的路径(一)

最新推荐文章于 2024-10-27 16:32:28 发布

原创最新推荐文章于 2024-10-27 16:32:28 发布 · 554 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#b树 #数据结构 #算法

剑指offer 专栏收录该内容

34 篇文章

订阅专栏

该博客介绍了一种算法，用于判断二叉树中是否存在从根节点到叶子节点的路径，其节点值之和等于给定的目标值。通过前序遍历的方法，逐层检查节点值，当到达叶子节点时判断路径和是否等于目标值。如果找到符合条件的路径，则返回true，否则返回false。

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

描述

给定一个二叉树root和一个值 sum ，判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。

1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点

2.叶子节点是指没有子节点的节点

3.路径只能从父节点到子节点，不能从子节点到父节点

4.总节点数目为n.

思路

按照前序遍历遍历接点，把每个节点的值都添加到count中，当走到叶子节点时，如果count == sum就返回true，否则就返回到父节点，并且将count的值减去刚刚加入进去的值。

代码

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return bool布尔型
     */
    boolean res = false;
    public boolean hasPathSum (TreeNode root, int sum) {
       
        if(sum == 0 || root == null){
            return false;
        }
       
       func(root,0,sum);
       return res;
       
    }
    public void func(TreeNode root, int count,int sum){
        if(res){
            return;
        }
        count = count + root.val;
        if(root.left != null && count != sum){
           func (root.left,count,sum); 
        }
         if(root.right != null && count != sum){
            func (root.right,count,sum); 
        }
        if(count == sum && root.left == null && root.right == null){
            //标致位，说明该路径已经找到了。
            res = true;
            return;
        }else{
            //走到叶子节点后，没有找到就结束，返回到父节点。
            return;
        }
        
    }
}