[剑指OFFER]之树：JZ34 二叉树中和为某一值的路径(二)

1.题目

输入一颗二叉树的根节点root和一个整数expectNumber，找出二叉树中结点值的和为expectNumber的所有路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点，不能从子节点到父节点
4.总节点数目为n

2.解法

2.1 树的结构

 class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
        public TreeNode(int val) {
            this.val = val;
        }
    }

2.2 解法1

这道题并没有固定“从根节点开始”这一条件，因此需要考虑每个节点都当做根节点开始递归。 也就是说每个节点都需要从头开始捋一遍。

    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        if (root == null) return ans;
        ArrayList<Integer> path = new ArrayList<>();
        helper(ans, path, root, target);
        return ans;
    }

    public void helper(ArrayList<ArrayList<Integer>> ans, ArrayList<Integer> path, TreeNode node, int target) {
        if (node == null) return;

        path.add(node.val);

        if (node.left == null && node.right == null && target == node.val) {
            ans.add(new ArrayList<>(path));
            return;
        }


        helper(ans, new ArrayList<>(path), node.left, target - node.val);
        helper(ans, new ArrayList<>(path), node.right, target - node.val);
    }

2.3 解法2

使用栈来模拟，类似与回溯算法。
跟之前的优势在于，不需要每次传入一个新的ArrayList<Integer> path来存储路径，而是可以直接复用Stack<Integer> path。

   public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        if (root == null) return ans;
        Stack<Integer> path = new Stack<>();
        helper(ans, path, root, target);
        return ans;
    }

    public void helper(ArrayList<ArrayList<Integer>> ans, Stack<Integer> path, TreeNode node, int target) {

        if (node == null) return;
        if (node.left == null && node.right == null && target == node.val) {
            path.push(node.val);
            ans.add(new ArrayList<>(path));
            return;
        }

        path.push(node.val);
        helper(ans, path, node.left, target - node.val);
        path.pop();

        path.push(node.val);
        helper(ans, path, node.right, target - node.val);
        path.pop();
    }

总结

需要考虑每个节点都当做根节点开始递归。 也就是说每个节点都需要从头开始捋一遍。

算法系列在github上有一个开源项目，主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客，将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。