【剑指offer】二叉树中和为某一值的路径

题目

        输入一棵二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

分析

     几个要点

        1、由于路径是从根节点出发到叶节点，也就是说路径总是以根节点为起始节点，因此我们需要遍历根节点。在树的前序、后序、中序遍历中，只有前序遍历是首先访问根节点的。

        2、由于二叉树没有指向父节点的指针，我们不知道前面经过了哪些节点，除非把经过路径上的节点保存下来，每访问一个节点，我们就把当前节点添加到路径中。

        3、利用链表存储结点，在该结点完成左右子树的路径搜索后（即递归函数结束，返回到其父结点之前），要删除该结点，从而记录别的路径。

     总体思路

        当用前序遍历的方式访问某个节点的时候，我们把该节点添加到某个路径上，并累加该节点的值。如果该节点是叶子节点，并且路径中节点的和等于输入的整数，则当前路径符合要求，我们将它打印出来。如果当前节点不是叶子节点，则继续访问它的子节点。当前节点访问结束之后，递归函数将自动回到它的父节点。因此在函数退出之前要在路径上删除当前节点，并减去当前节点的值，以确保返回父节点时的路径正好是从根节点到父节点。

        保存路径的数据结构本质上是一个栈，路径要和递归调用状态一致，而递归调用的本质就是压栈和出栈的过程。

代码

        

package com.TreeSerial;

import java.util.ArrayList;

public class FindPathInBinaryTree2 {
    public static class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;

        public TreeNode(int val){
            this.val = val;
        }
    }

    static ArrayList<ArrayList<Integer>> res = new ArrayList<>();
    static ArrayList<Integer> path = new ArrayList<>();

    public static ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target){
        if(root == null){
            return res;
        }
        path.add(root.val);
        target -= root.val;
        //如果是叶子结点，并且路径上的节点值的和等于输入的值，则打印出这条路径
        if(target == 0 && root.left == null && root.right == null) {
            res.add(new ArrayList<Integer>(path));
        }
        //如果不是叶子结点，就遍历它的左右节点
        FindPath(root.left, target);
        FindPath(root.right, target);
        path.remove(path.size()-1);//返回父节点前，在路径上删除当前节点

        return res;

    }

    public static void main(String[] args){
        TreeNode tn = new TreeNode(10);
        tn.left = new TreeNode(5);
        tn.left.left = new TreeNode(4);
        tn.left.right = new TreeNode(7);
        tn.right = new TreeNode(12);
        ArrayList<ArrayList<Integer>> res = FindPath(tn, 22);
        System.out.println(res);

        for(ArrayList<Integer> path : res) {
            System.out.println(path);
        }
    }
}