欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

二叉树中和为值的路径

程序员文章站 2022-05-21 19:26:32
...
  • 题目:
    输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
  • 分析:
    借助ArrayList< ArrayList < Integer >>存储多条路径。内部链表为arrList,外部链表为arrListAll。
    1.当前遍历节点为空,直接返回arrListAll;当前节点不为空,当前节点值假如arrList;目标值target减去改节点的值.
    2.若此时路径之和等于该目标值,即target==0,左右节点都为空,将该路径加入arrListAll中.
    3.递归遍历节点的左右子节点,直至某左右节点都为空,该路径不存在,则从arrList删除,回到其父节点
  • 代码:
public class Solution {
    ArrayList<ArrayList<Integer>> arrListAll = new ArrayList<ArrayList<Integer>>();
    ArrayList<Integer> arrList = new ArrayList<Integer>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null)
            return arrListAll;
        arrList.add(root.val);
        target -= root.val;
        if(target == 0 && root.left == null && root.right == null)
            arrListAll.add(new ArrayList<Integer>(arrList));
        FindPath(root.left, target);
        FindPath(root.right, target);
        arrList.remove(arrList.size()-1);
        return arrListAll;
    }
}

方法二、
利用栈实现:
参考:https://blog.csdn.net/jsqfengbao/article/details/47291207