剑指Offer二叉树和为某一值的路径

最新推荐文章于 2024-12-16 21:23:07 发布

原创最新推荐文章于 2024-12-16 21:23:07 发布 · 138 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

牛客网专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一种算法，该算法通过递归先序遍历二叉树，找到所有节点值之和等于指定整数的路径。路径定义为从根节点到叶节点的节点序列。文章详细解释了算法的实现过程，包括使用栈来保存遍历路径，以及在达到叶节点时检查路径总和是否匹配目标值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

/**
这道题的思路是通过递归的先序遍历，保存每一次遍历的数据，然后当到达叶子节点的时候，
对它的和进行判断，如果符合的话，那就把保存下来，然后接着返回父节点，并把当前节点从
路径中删除
*/

import java.util.ArrayList;
import java.util.Stack;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
          
              
        ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
        if(root==null)
            return lists;
		Stack<Integer> stack = new Stack<>();
		findPath(root, target, 0, stack, lists);
		return lists;
    }
   public void findPath(TreeNode root, int target, int currentSum, Stack<Integer> stack,
			ArrayList<ArrayList<Integer>> lists) {
		currentSum += root.val;
		stack.push(root.val);
		if (root.left == null && root.right == null && currentSum == target) {
			int sum = 0;
			ArrayList<Integer> list = new ArrayList<>();
			for (int i = 0; i < stack.size(); i++) {

				list.add(stack.get(i));
			}
			lists.add(list);
		}
		if (root.left != null)// 当左子树不为空的时候，寻找左子树
			findPath(root.left, target, currentSum, stack, lists);
		if (root.right != null)// 当右子树不为空的时候，寻找右子树
			findPath(root.right, target, currentSum, stack, lists);
		// 当返回父节点时，在路径上删除当前节点
		stack.pop();

	}
}