leetcode刷题:二叉树16(路径总和)

71 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

112. 路径总和

力扣题目链接

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

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

示例: 给定如下二叉树,以及目标和 sum = 22,

112.路径总和1

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

递归遍历,如果只有左子树,则只计算左子树,如果只有右子树,则只计算右子树,遇到空节点则开始比较,把递归结果 或 运算 即可。

package com.programmercarl.tree;

import com.programmercarl.util.GenerateTreeNode;

import java.util.Stack;

/**
 * @ClassName HasPathSum
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/5 11:40
 * @Version 1.0
 * https://leetcode.cn/problems/path-sum/
 * 112. 路径总和
 **/
public class HasPathSum {
    /**
     * 给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
     * 叶子节点 是指没有子节点的节点。
     * <p>
     * ..不就是遍历每条路径和嘛
     *
     * @param root
     * @param targetSum
     * @return
     */
    public boolean hasPathSum(TreeNode root, int targetSum) {
        //使用递归
        if (root == null) {
            return false;
        }
        return traversal(root, 0, targetSum);
    }

    public boolean traversal(TreeNode root, int sum, int targetSum) {
        if (root == null) {
            return sum == targetSum;
        }
        //只计算到叶子结点
        sum += root.val;
        //只计算右子树的结果
        if (root.left == null && root.right != null) {
            return traversal(root.right, sum, targetSum);
        }
        //只计算左子树的结果
        if (root.left != null && root.right == null) {
            return traversal(root.left, sum, targetSum);
        }
        return traversal(root.left, sum, targetSum) || traversal(root.right, sum, targetSum);
    }

    public static void main(String[] args) {
        System.out.println(new HasPathSum().hasPathSum(GenerateTreeNode.generateTreeNode("[5,4,8,11,null,13,4,7,2,null,null,null,1]"), 22));
    }
}

在这里插入图片描述