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

	private void FindPath(TreeNode root, int target, Stack stack, ArrayList> pathList) {
		
		if (root == null) {
			return;
		}
		
		//如果当前节点是叶子节点
		if (root.left == null && root.right == null) {
			
			if (root.val == target) {
				//如果当前节点的值和目标值相等
				ArrayList path = new ArrayList<>();
				//1.遍历栈,将此时栈中的元素全部加入到当前这条路径列表中
				Iterator 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();
		}
		
	}

}


你可能感兴趣的:(剑指offer面试题,剑指offer,面试题,算法)