二叉树中和为某一值的路径

最新推荐文章于 2025-03-25 10:51:28 发布

原创最新推荐文章于 2025-03-25 10:51:28 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

算法学习同时被 2 个专栏收录

34 篇文章

订阅专栏

JAVA

13 篇文章

订阅专栏

本文介绍了一种用于寻找二叉树中所有从根节点到叶子节点且节点值之和等于特定数值的路径的算法。该算法通过递归方式遍历二叉树的每个节点，并使用栈来记录路径上的节点值。

/**
 * 二叉树中和为某一值的路径   从树的根节点开始往下一直到叶结点所经过的结点形成的路径
 * @author Q.Yuan
 *
 */
public class BinaryTreePath {
	private Scanner scanner = new Scanner(System.in);
	//先序建立二叉树,测试数据10 5 4 # # 7 # # 12 # #
	public BTreeNode createTree(BTreeNode root){
		String data = scanner.next();
		if(data.equals("#"))
			return null;
		root = new BTreeNode();
		root.data = Integer.valueOf(data);
		root.left = createTree(root.left);
		root.right = createTree(root.right);
		return root;
	}
	/**
	 * 
	 * @param root  树的根节点
	 * @param sum   当前路径之和
	 * @param stack 存放路径上结点的值
	 * @param num   求的路径和
	 */
	public void printPath(BTreeNode root,int sum,Stack<Integer> stack,int num){
		if(root != null){
			sum = sum + root.data;
			stack.push(root.data);
			printPath(root.left, sum, stack, num);
			printPath(root.right, sum, stack, num);
			if(sum == num && root.left == null && root.right == null){
				for(int i:stack){
					System.out.print(i+" ");
				}
				System.out.println();
			}
			stack.pop();
		}
	}
	
	public static void main(String[] args) {
		BinaryTreePath btp = new BinaryTreePath();
		BTreeNode root = null;
		root = btp.createTree(root);
		Stack<Integer> stack = new Stack<Integer>();
		int num = 22;
		int sum = 0;
		btp.printPath(root, sum, stack, num);
	}
}
//树的结构
class BTreeNode{
	Integer data;
	BTreeNode left;
	BTreeNode right;
}