题目:
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。
从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
其实就是树的遍历。
在遍历的同时记录下来轨迹。
打印的时候需要查看轨迹:从头到尾
往下走的时候需要往轨迹的末尾加:
往上反的时候需要往轨迹的末尾减:
所以这里使用LinkedList把。。
树的定义如下:
package com.aii.algorithm;
public class TreeNode {
int value;
TreeNode left;
TreeNode right;
@Override
public String toString() {
return "TreeNode [value=" + value + "]";
}
public TreeNode(int value) {
super();
this.value = value;
}
}
实现代码:
package com.aii.algorithm;
import java.util.LinkedList;
public class FindPath {
public void find(TreeNode root, int target, LinkedList<TreeNode> list) {
if (root == null) {
return;
}
if (list == null) {
list = new LinkedList<TreeNode>();
}
// 得到当前值
int currentValue = root.value;
// 那么剩下的目标值就可以减得到
target = target - currentValue;
// 当前为叶子节点的情况-
if (root.left == null && root.right == null) {
// 该路径是,打印
if (target == 0) {
print(list, root);
return;
}
// 否则,结束
return;
}
// 放入容器中记录,,找到路径的时候要用
list.offer(root);
// 递归
find(root.left, target, list);
find(root.right, target, list);
// 如果次路径没找到,则把刚放入的取出,
list.removeLast();
}
private void print(LinkedList<TreeNode> list, TreeNode root) {
for (TreeNode tr : list) {
System.out.print(tr);
System.out.print(",");
}
System.out.println(root);
}
}