前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >路径总和

路径总和

作者头像
你的益达
发布2020-08-05 15:35:39
6580
发布2020-08-05 15:35:39
举报

路径总和

问题描述:

给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

代码语言:javascript
复制
示例: 
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解决方案

修改二叉树深度优先遍历即可,不过需要注意的是叶子结点时没有子节点的结点。实现代码如下:

代码语言:javascript
复制
class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if(root == null){
            return false;
        }
        return dfs(root, sum);
    }
    public boolean dfs(TreeNode root, int sum){
        sum -= root.val;
        if(root.left == null && root.right == null){
            return sum == 0;
        }
        boolean ans = false;
        if(root.left != null){
            ans = ans || dfs(root.left, sum);
        }
        if(root.right != null){
            ans = ans || dfs(root.right, sum);
        }
        return ans;
    }
}

路径总和II

问题描述:

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

代码语言:javascript
复制
示例:
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解决方案

直接dfs。

代码语言:javascript
复制
class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root == null){
            return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        dfs(root, sum, new Stack<>(), result);
        return result;
    }
    public void dfs(TreeNode root, int sum, Stack<Integer> stack, List<List<Integer>> result){
        sum -= root.val;
        stack.push(root.val);
        if(root.left == null && root.right == null){
            if(sum == 0){
                result.add(new ArrayList<>(stack));
            }
            stack.pop();
            return;
        }
        if(root.left != null){
            dfs(root.left, sum, stack, result);
        }
        if(root.right != null){
            dfs(root.right, sum, stack, result);
        }
        stack.pop();
    }
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-07-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 路径总和
    • 问题描述:
      • 解决方案
      • 路径总和II
        • 问题描述:
          • 解决方案
          领券
          问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档