【LeetCode-中等题】437. 路径总和 III

题目

在这里插入图片描述

方法一:迭代层序 + 每层节点dfs 维护一个count变量

思路:

  1. 层序遍历每一个节点
  2. 遍历一个节点就对这个节点进行dfs
  3. dfs的同时,维护一个count变量,并且把每次路径节点进行累加,然后判断sum==target 如果相等 让count++
  4. 直到当前节点的dfs结束后,再进行下一个节点的dfs
  5. 最后输出count
    在这里插入图片描述
// 方法一 迭代层序 + 每层节点dfs  维护一个count变量
    int count ;
    public int pathSum(TreeNode root, int targetSum) {
    
    
    if(root == null) return 0;
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root);
    while(!queue.isEmpty()){
    
    
        int size = queue.size();
        for(int i =0 ;i<size;i++){
    
    
          TreeNode node = queue.poll();
          checkPathDfs(node,0,targetSum);//对该节点的dfs的同时  累加路径的值同时和target对比,若sum累加值 == target 则count++
          if(node.left != null) queue.offer(node.left);
          if(node.right != null) queue.offer(node.right);
        }  
    }
     return count;
    }
    //dfs:累加路径的值同时和target对比,若sum累加值 == target 则count++
    public void checkPathDfs(TreeNode root, long sum,int targetSum) {
    
    //注意  sum要为long  因为测试案例超出了int的范围了  sum值会处在封顶的位置,而不是真实值
      if(root == null) return ;
      sum = sum + root.val;
      if(sum == targetSum) count++;
      checkPathDfs(root.right,sum,targetSum);//对右子树dfs 并且sum继承过来    左右子树的sum归左子树的,右子树的sum归右子树  互不干扰
      checkPathDfs(root.left,sum,targetSum);
    }

猜你喜欢

转载自blog.csdn.net/weixin_45618869/article/details/132600804
今日推荐