牛客在线编程101-29 二叉树中和为某一值的路径(一)

客官°小女子只卖身不卖艺 2023-10-04 20:11 1阅读 0赞

描述  
给定一个二叉树root和一个值 sum ，判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。  
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点  
2.叶子节点是指没有子节点的节点  
3.路径只能从父节点到子节点，不能从子节点到父节点  
4.总节点数目为n

例如：  
给出如下的二叉树，\\ sum=22 sum=22，

返回true，因为存在一条路径 5\\to 4\\to 11\\to 25→4→11→2的节点值之和为 22

数据范围：  
1.树上的节点数满足 0 \\le n \\le 100000≤n≤10000  
2.每 个节点的值都满足 |val| \\le 1000∣val∣≤1000  
要求：空间复杂度 O(n)O(n)，时间复杂度 O(n)O(n)  
进阶：空间复杂度 O(树的高度)O(树的高度)，时间复杂度 O(n)O(n)

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) {
        
            // write code here
            if(root == null) {
        
                return res;
            }
            dfs(root, 0, sum);
            return res;
        }
    
        public void dfs(TreeNode cur, int cur_sum, int sum) {
        
            if (cur == null) {
        
                return;
            }
    
            cur_sum += cur.val;
            if (cur.left == null && cur.right == null && cur_sum == sum) {
        
                res = true;
                return;
            }
    
            if (cur.left != null) {
        
                dfs(cur.left, cur_sum, sum);
            }
    
            if (cur.right != null) {
        
                dfs(cur.right, cur_sum, sum);
            }
        }
    }