题目
题目:输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。路径是从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
注意:路径必须是从根节点开始一直到叶节点才可以。。。刚开始一直没有注意到这一点。。
思路
打印出二叉树中节点值的和为输入整数的所有路径,当访问到叶节点的时候,我们不知道前面经了哪些节点,因此我们需要一个集合来存储经过的节点,(该集合必须是先进后出)
路径必须是从根节点开始一直到叶节点才可以,因此我们按照前序遍历的方式,从根节点一直遍历到其某一叶节点,然后判断其和是否与要求的和相等,如果相等则打印该路径上的所有节点值;
如果不相等,或者打印完成之后,然后从集合中删除该节点,返回其父节点,继续递归判断
java代码
package com.ca;
import java.util.ArrayDeque;
//定义节点
class Node{
int value;
Node leftTree;
Node rightTree;
public Node(int value){
this.value = value;
}
@Override
public String toString(){
return this.value + "";
}
}
public class SumFixedValue {
public static void main(String[] args) {
Node root = new Node(8);
root.leftTree = new Node(2);
root.leftTree.leftTree = new Node(5);
root.leftTree.rightTree = new Node(9);
root.rightTree = new Node(3);
root.rightTree.rightTree = new Node(4);
int sum = 15;
int CurrentSum = 0;
ArrayDeque<Node> ad = new ArrayDeque<Node>();
Search(root,ad,sum,CurrentSum);
}
public static void Search(Node root,ArrayDeque<Node> ad,int sum,int currentSum){
if(root == null){
return;
}
currentSum += root.value;
ad.push(root);
//判断是否为叶节点,如果是叶节点,判断其和是否与sum相同,如果相同则输出
boolean is = (root.leftTree == null) && (root.rightTree == null);
if(currentSum == sum && is){
System.out.println(ad);
}
//如果不是叶节点,遍历他的左右节点
if(root.leftTree != null){
Search(root.leftTree,ad,sum,currentSum);
}
if(root.rightTree != null){
Search(root.rightTree,ad,sum,currentSum);
}
//在返回父节点之前,删除当前节点
ad.pop();
}
}
扩展
如果题目中没有限定路径必须是从根节点开始一直到叶节点结束呢,不管从哪开始到哪结束,只要该路径上节点和为要求值,该怎么做??
没有想到。。。