[数据结构与算法]LeetCode_112_路径总和

题目链接

• https://leetcode.cn/problems/path-sum/

题目描述

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径，这条路径上所有节点值相加等于目标和 targetSum 。如果存在，返回 true ；否则，返回 false 。

叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出：true
解释：等于目标和的根节点到叶节点路径如上图所示。

示例 2：

输入：root = [1,2,3], targetSum = 5
输出：false
解释：树中存在两条根节点到叶子节点的路径：
(1 --> 2): 和为 3
(1 --> 3): 和为 4
不存在 sum = 5 的根节点到叶子节点的路径。

示例 3：

输入：root = [], targetSum = 0
输出：false
解释：由于树是空的，所以不存在根节点到叶子节点的路径。

提示：

• 树中节点的数目在范围 [0, 5000] 内
• -1000 <= Node.val <= 1000
• -1000 <= targetSum <= 1000

解题思路

递归法

• 确认递归函数的参数和返回类型
  • 参数：根节点root，计数器count：用来计算一条边之和是否正好是target值
  • 返回类型：boolean
• 确认终止条件
  • 初始时count = target，然后每次减去遍历路径节点上的数值，如果最后count == 0且同时到了叶子节点的话，就说明找到了目标和。如果遍历到了叶子节点，但是count != 0，说明没有符合的路径
• 确认单层逻辑
  • 因为终止条件是判断叶子节点，所以递归的过程中就不用让空节点进入递归。
  • 此外，递归函数是有返回值的，如果递归函数return true，说明找到了合适的路径，应该立刻返回

迭代法

使用两个栈stack1用来进行遍历，stack2用来进行数值相加，判断能否找到满足要求的路径

AC代码

递归法

class Solution {
    public boolean hasPathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return false;
        }
        targetSum -= root.val;
        if (root.left == null && root.right == null) {
            return targetSum == 0;
        }
        if (root.left != null) {
            boolean left = hasPathSum(root.left, targetSum);
            if (left) {
                return true;
            }
        }
        if (root.right != null) {
            return hasPathSum(root.right, targetSum);
        }
        return false;
    }
}

迭代法

class Solution {
    public boolean hasPathSum(TreeNode root, int targetSum) {
                if (root == null) {
            return false;
        }
        Stack<TreeNode> stack1 = new Stack<>();
        Stack<Integer> stack2 = new Stack<>();
        stack1.push(root);
        stack2.push(root.val);
        while (!stack1.isEmpty()) {
            int len = stack1.size();
            for (int i = 0; i < len; i++) {
                TreeNode tmp = stack1.pop();
                int sum = stack2.pop();
                if (tmp.left == null && tmp.right == null && sum == targetSum) {
                    return true;
                }
                if (tmp.left != null) {
                    stack1.push(tmp.left);
                    stack2.push(sum + tmp.left.val);
                }
                if (tmp.right != null) {
                    stack1.push(tmp.right);
                    stack2.push(sum + tmp.right.val);
                }
            }
        }
        return false;
    }
}