题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路
首先思考节点值的和为输入的整数,每条路径都一定是从根节点到叶子节点,在数据结构中从根节点到叶子节点的遍历称之为深度优先遍历DFS。因此整个过程可以采用先序遍历方式的DFS,即根节点》左子树》右子树。随后考虑一次遍历完成后的处理,当一次遍历完成后,如果输入整数值恰好等于节点值之和,则输出这条路径并且回退一个节点;如果不等于则直接回退一个节点,即回退到当前节点的父节点,如果该父节点有右孩子,则继续遍历,否则继续回退。考虑回退到根节点,此时如果它有右孩子,则继续遍历,否则整个DFS结束。
数据结构分析:使用java完成整个题目,采用ArrayList<ArrayList<Integer>>,举个如下的例子,输入整数为22。
第一次遍历 10,5,4 19<22,因此ArrayList要使用remove函数回退到父节点5上,开始寻找右孩子。
第二次遍历 10,5,7 22=22,此时正好输出这条路径(将路径添加到ArrayList中),10,5,7,回退到父节点5上,此时5的左右孩子都已遍历完成,因此回退到父节点10
第三次遍历 10,12 22=22,此时正好输出这条路径(将路径添加到ArrayList中),10,12,由于12没有左右孩子,因此直接回退到父节点10
由于根节点10左右孩子都已遍历完成,因此整个DFS结束。
需要注意的是不论路径的值是否等于输入整数值,都要回退,即使用remove函数移除路径上的最后一个节点。
代码实现
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> arr = new ArrayList<Integer>();
int sum = 0;
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
//1.判断节点是否为空
if(root == null)
return list;
//2.将当前节点加入至路径 arr中
arr.add(root.val);
sum +=root.val;
//3.判断当前节点是否为叶子节点
boolean isLeaf = (root.left==null)&&(root.right==null);
//4.若当前路径权值和为target,保存该路径所有节点
if(sum == target &&isLeaf){
ArrayList<Integer> now = new ArrayList<Integer>();
for(int i=0;i<arr.size();i++){
now.add(arr.get(i));
}
list.add(now);
}
//5.分别递归左子树和右子树
if(sum<target&&root.left!=null){
FindPath(root.left,target);
}
if(sum<target&&root.right!=null){
FindPath(root.right,target);
}
arr.remove(arr.size()-1);
sum -= root.val;
return list;
}
}