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