Leetcode 112. 路径总和

题目

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

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

来自力扣：leetcode 112. 路径总和

解答

解法一：递归

按照前序遍历的顺序递归。

当到达叶子结点的时候，判断此时 sum 是否能被减成 0。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if(root == null) return false;
        if(root.left == null && root.right == null) return sum == root.val;
        return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);
    }
}

结果

解法二：深度搜索+前序遍历

其实就是模拟前序遍历的非递归版。

使用 栈 数据结构，注意 先压右结点，后压左结点 即可。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

class Solution {
    
    class Pair {
        TreeNode node;
        int need;
        
        Pair(TreeNode node, int need) {
            this.node = node;
            this.need = need;
        }
    }
    
    public boolean hasPathSum(TreeNode root, int sum) {
        if(root == null) return false;
        
        LinkedList<Pair> stack = new LinkedList<>();         
        stack.push(new Pair(root, sum));
        while(!stack.isEmpty()) {
            Pair pair = stack.pop();
            TreeNode node = pair.node;
            int need = pair.need;
            
            if(node.left == null && node.right == null && need == node.val) {
                return true;
            }
            
            if(node.right != null) {
                stack.push(new Pair(node.right, need - node.val));
            }
            
            if(node.left != null) {
                stack.push(new Pair(node.left, need - node.val));
            }
        }
        
        return false;
    }
}

结果