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

来源:互联网 发布:linux下echo命令 编辑:程序博客网 时间:2024/05/29 16:54
import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;


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


    }


解题思路:


由于路径是从根结点出发到叶结点, 也就是说路径总是以根结点为起始点,因此我们首先需要遍历根结点。


在树的前序、中序、后序三种遍历方式中,只有前序遍历是首先访问根结点的。
当用前序遍历的方式访问到某一结点时,用目标路径减去当前结点的值。如果为0,并且当前结点为叶子结点,则打印。
否则递归判断当前结点的左右结点。


当前结点访问结束后,递归函数将自动回到它的父结点。
因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,
以确保返回父结点时路径刚好是从根结点到父结点的路径。


不难看出保存路径的数据结构实际上是一个枝, 因为路径要与递归调用状态一致, 而递归调用的本质就是一个压栈和出栈的过程。
*/
public class Solution {
    public ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
    private ArrayList<Integer> list = new ArrayList<Integer>();
    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null) return listAll;
        list.add(root.val);
        target =target - root.val;
        
        if(target == 0 && root.left == null && root.right == null){
            listAll.add(new ArrayList<Integer>(list));
        }
        FindPath(root.left, target);
        FindPath(root.right, target);
        
        list.remove(list.size()-1);//比如10 5 4 不满足目标值,得除去4.。还得考虑不满足叶子结点  
        return listAll;
    }
    /*
       10
        //         /      \
        //        5        12
        //       /\
        //      4  7
      if(root == null) return listAll;
         int curSum = 0;
         if (root != null) {
            // 加上当前结点的值
            curSum += root.val;
            // 将当前结点入队
            list.add(root.val);
            // 如果当前结点的值小于期望的和
            if (curSum < target) {
                // 递归处理左子树
                FindPath(root.left,target);
                // 递归处理右子树
                FindPath(root.right,target);
            }
            // 如果当前和与期望的和相等
            else if (curSum == target) {
                // 当前结点是叶结点,则输出结果
                if (root.left == null && root.right == null) {
                    listAll.add(new ArrayList<Integer>(list));
                }
            }
            // 移除当前结点
            list.remove(list.size() - 1);
        }
        return listAll;
    */
}
原创粉丝点击