剑指offer面试题25 二叉树中和为某一值的路径

解题思路:

首先需要理解路径的概念:路径总是从根节点到叶节点,由于路径是从根节点开始,所以可以考虑用先序遍历的方式。每遍历完一个节点进入节点的左子树时,需要将当前节点保存起来,作为路径中的一个节点,之所以需要保存路径上的节点,是因为需要计算路径的和,是否和待查找元素相同。因此考虑用栈来保存路径。注意:当从子节点返回父节点时,需要将当前元素出栈。用于更新路径。

import java.util.ArrayList;
import java.util.Iterator;
import java.util.Stack;

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>> pathList = new ArrayList<>();
		if (root == null) {
			
			return pathList;
			
		}
		//创建一个栈,用于保存从根节点到叶节点的路径
		Stack<Integer> stack = new Stack<>();
		FindPath(root, target,stack,pathList);
		return pathList;
	}

	private void FindPath(TreeNode root, int target, Stack<Integer> stack, ArrayList<ArrayList<Integer>> pathList) {
		
		if (root == null) {
			return;
		}
		
		//如果当前节点是叶子节点
		if (root.left == null && root.right == null) {
			
			if (root.val == target) {
				//如果当前节点的值和目标值相等
				ArrayList<Integer> path = new ArrayList<>();
				//1.遍历栈,将此时栈中的元素全部加入到当前这条路径列表中
				Iterator<Integer> iterator = stack.iterator();
				while (iterator.hasNext()) {
					path.add(iterator.next()); 
				}
				//2.将当前节点加入列表中
				path.add(root.val);
				//3.将符合条件的路径加入路径列表中
				pathList.add(path);
			}
			
		} else {
			stack.push(root.val);
			//对左子树查找值为某一个数的路径
			FindPath(root.left, target - root.val, stack, pathList);
			//对右子树查找值为某一个数的路径
			FindPath(root.right, target - root.val, stack, pathList);
			//当前节点不是叶子节点,在返回上一层父节点时,要将当前子节点出栈
			stack.pop();
		}
		
	}

}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值