剑指offer解题报告(Java版)——二叉树中和为某一值的路径 25
分析问题
首先需要明白路径的概念，路径是从根节点出发，一直到叶子节点，所形成的一条路径
所以我们寻找和为某一值的路径的时候，首先从根节点10出发，前序遍历二叉树，路径为10,5,4，发现和不为22，所以需要从4返回到5，然后再走7，发现和为22，记录下这条路径
然后再返回到5，再返回到10，再从10的右子树开始，到12，又记录下一条路径
发现这是一个栈的结构，所以我们考虑用一个栈来存储路径，另外我们需要一个结果来存储currentSum，用于与exceptedSum相比较
解决问题
public void findPath(BinaryTreeNode root, int expectedSum,
Stack<Integer> path, int currentSum) {
if (root == null)
return;
currentSum += root.data;
path.push(root.data);
boolean isLeaf = root.leftNode == null && root.rightNode == null;
if (isLeaf) {
if (currentSum == expectedSum) {
System.out.println("A path is");
for (int i : path)
System.out.print(i+" ");
System.out.println();
}
} else {
if (root.leftNode != null) {
findPath(root.leftNode, expectedSum, path, currentSum);
}
if (root.rightNode != null) {
findPath(root.rightNode, expectedSum, path, currentSum);
}
}
currentSum -= root.data;
path.pop();
}
测试代码
MySo2 mySo2=new MySo2();
Stack path=new Stack<>();
int currentSum=0;
mySo2.findPath(root1, 22, path, currentSum);
首先创建一个stack用于存放路径，作为参数传入findpath中，另外也将currentSum作为参数传入
在findPath函数中，首先判断一下root是否为空，为空的话就没什么好玩儿的了
不为空，那么将currentSum加上root的值，表明现在的Sum，将root的值压入栈中，这里取了个巧，只压入了root的data值，而不是压入的节点
因为这里仅仅需要实现的一个功能是在递归返回上一层的时候，将节点从路径中删除，而返回上一层节点在递归的时候就已经实现了
对于路径来说，需要判断是否抵达叶子节点，叶子节点的定义是没有左右孩子，这里将对节点的判断用个boolean值来替代，使得代码简洁一些
如果是叶子节点，判断一下currentSum和exceptedSum是否相等，如果相等，将路径打印出来
如果不是叶子节点，那么就需要前序遍历递归到左子树，然后返回到上一层，递归到右子树
右子树返回的时候，需要将栈弹出，说明要将这个节点从路径中删除
实例说明
说一下上述树的递归过程吧，首先传入10，currentSum=10，将10压入栈中，10不是叶子节点，有左孩子5，进入到左子树，也就是递归的下一层,这时候的参数root=5，exceptedSum一直没变是22，这里后面不再说了，path中有10，currentSum=10，传进来之后，5不为空（其实这里每次都要判读一下不是很好，可以优化），currentSum要加上这个5，成为15，5压入栈中，5也不是叶子节点，继续进入左子树，为4，currentSum加上4，为19，4压入栈中，4是叶子节点，但是19不等于22，这里不会进入else了，就直接currentSum减去4，并将栈中的4弹出，此时栈中只有10和5，进入到5的右子树，同样的判断，和为22，打印出来，然后弹出7，返回到上一层，由于上一层已经执行到进入右子树那条语句，也就是if (root.rightNode != null) findPath(root.rightNode, expectedSum, path, currentSum);所以下一条应该执行的语句就是currentSum减去5，将5弹出，此时再返回上一层，也就是现在栈中只有10，由于在10这一层，只执行到
if (root.leftNode != null) findPath(root.leftNode, expectedSum, path, currentSum);这条语句，接下来应该执行进入右子树的语句了，接下来也就没什么好说的了
